diff --git a/.output/1.txt b/.output/1.txt
deleted file mode 100644
index 56a6051ca2b02b04ef92d5150c9ef600403cb1de..0000000000000000000000000000000000000000
--- a/.output/1.txt
+++ /dev/null
@@ -1 +0,0 @@
-1
\ No newline at end of file
diff --git a/Dockerfile b/Dockerfile
deleted file mode 100644
index fa21a3c442ca872c1c030975e07ec7f170b3be0b..0000000000000000000000000000000000000000
--- a/Dockerfile
+++ /dev/null
@@ -1,98 +0,0 @@
-<<<<<<< HEAD
-# Step 1: Use an official Python slim base image
-FROM python:3.10-slim
-
-# Step 2: Install system dependencies
-RUN apt-get update && apt-get install -y \
- wget \
- tar \
- && apt-get clean
-
-# Step 3: Add a non-root user (required by Hugging Face Spaces)
-RUN useradd -m -u 1000 user
-
-# Step 4: Switch to the "user" user
-USER user
-
-# Step 5: Set home and working directory
-ENV HOME=/home/user \
- PATH=/home/user/.local/bin:$PATH
-WORKDIR $HOME/app
-
-# Step 6: Copy requirements into the container
-COPY --chown=user requirements.txt ./requirements.txt
-
-# Step 7: Install Python dependencies
-RUN pip install --no-cache-dir --upgrade pip && \
- pip install --no-cache-dir -r requirements.txt
-
-# Step 8: Copy all necessary files and folders into the container
-COPY --chown=user .output ./.output
-COPY --chown=user cache ./cache
-COPY --chown=user input ./input
-COPY --chown=user output ./output
-COPY --chown=user prompts ./prompts
-COPY --chown=user reports ./reports
-COPY --chown=user auth.py ./auth.py
-COPY --chown=user knowledge_graph.html ./knowledge_graph.html
-COPY --chown=user query_config.yaml ./query_config.yaml
-COPY --chown=user app.py ./app.py
-COPY --chown=user search_handlers.py ./search_handlers.py
-COPY --chown=user settings.yaml ./settings.yaml
-COPY --chown=user styles.css ./styles.css
-COPY --chown=user wiki.py ./wiki.py
-
-# Step 10: Expose the Streamlit default port
-EXPOSE 7860
-
-# Step 11: Define the entrypoint command
-=======
-# Step 1: Use an official Python slim base image
-FROM python:3.10-slim
-
-# Step 2: Install system dependencies
-RUN apt-get update && apt-get install -y \
- wget \
- tar \
- && apt-get clean
-
-# Step 3: Add a non-root user (required by Hugging Face Spaces)
-RUN useradd -m -u 1000 user
-
-# Step 4: Switch to the "user" user
-USER user
-
-# Step 5: Set home and working directory
-ENV HOME=/home/user \
- PATH=/home/user/.local/bin:$PATH
-WORKDIR $HOME/app
-
-# Step 6: Copy requirements into the container
-COPY --chown=user requirements.txt ./requirements.txt
-
-# Step 7: Install Python dependencies
-RUN pip install --no-cache-dir --upgrade pip && \
- pip install --no-cache-dir -r requirements.txt
-
-# Step 8: Copy all necessary files and folders into the container
-COPY --chown=user .output ./.output
-COPY --chown=user cache ./cache
-COPY --chown=user input ./input
-COPY --chown=user output ./output
-COPY --chown=user prompts ./prompts
-COPY --chown=user reports ./reports
-COPY --chown=user auth.py ./auth.py
-COPY --chown=user knowledge_graph.html ./knowledge_graph.html
-COPY --chown=user query_config.yaml ./query_config.yaml
-COPY --chown=user app.py ./app.py
-COPY --chown=user search_handlers.py ./search_handlers.py
-COPY --chown=user settings.yaml ./settings.yaml
-COPY --chown=user styles.css ./styles.css
-COPY --chown=user wiki.py ./wiki.py
-
-# Step 10: Expose the Streamlit default port
-EXPOSE 7860
-
-# Step 11: Define the entrypoint command
->>>>>>> b96b49ad31dd18a91ca19d59316e1b3ef2f531ff
-CMD ["streamlit", "run", "app.py", "--server.port", "7860", "--server.address", "0.0.0.0"]
\ No newline at end of file
diff --git a/README.md b/README.md
deleted file mode 100644
index d2978015280a14c42976023ca830e0112e5e9caf..0000000000000000000000000000000000000000
--- a/README.md
+++ /dev/null
@@ -1,10 +0,0 @@
----
-title: PwcGraphRAG
-emoji: ⚡
-colorFrom: purple
-colorTo: yellow
-sdk: docker
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
diff --git a/__pycache__/auth.cpython-311.pyc b/__pycache__/auth.cpython-311.pyc
deleted file mode 100644
index 41f41dcf41ec40ae71710ee4b261bf84ff79f54c..0000000000000000000000000000000000000000
Binary files a/__pycache__/auth.cpython-311.pyc and /dev/null differ
diff --git a/__pycache__/search_handlers.cpython-311.pyc b/__pycache__/search_handlers.cpython-311.pyc
deleted file mode 100644
index 81192312e1187c065c141dc26aa9a0ae1c27cbb4..0000000000000000000000000000000000000000
Binary files a/__pycache__/search_handlers.cpython-311.pyc and /dev/null differ
diff --git a/__pycache__/wiki.cpython-311.pyc b/__pycache__/wiki.cpython-311.pyc
deleted file mode 100644
index 3273d5e0a939dbdddfc86f507cc9e84b4022ce45..0000000000000000000000000000000000000000
Binary files a/__pycache__/wiki.cpython-311.pyc and /dev/null differ
diff --git a/app.py b/app.py
deleted file mode 100644
index 01fdc2587c75a784bb5afd86b5d2246c80f34717..0000000000000000000000000000000000000000
--- a/app.py
+++ /dev/null
@@ -1,613 +0,0 @@
-<<<<<<< HEAD
-import streamlit as st
-import asyncio
-import sys
-from pathlib import Path
-import base64
-import pandas as pd
-from typing import Literal, Tuple, Optional
-from wiki import render_wiki_tab
-from search_handlers import run_global_search, run_local_search, run_drift_search
-import auth
-
-
-import graphrag.api as api
-from graphrag.config import GraphRagConfig, load_config, resolve_paths
-from graphrag.index.create_pipeline_config import create_pipeline_config
-from graphrag.logging import PrintProgressReporter
-from graphrag.utils.storage import _create_storage, _load_table_from_storage
-
-
-st.set_page_config(page_title="GraphRAG Chat Interface", page_icon="🔍", layout="wide")
-
-# Define default avatars at the module level
-DEFAULT_USER_AVATAR = "👤"
-DEFAULT_BOT_AVATAR = "🤖"
-
-# Initialize session state for avatars
-if "user_avatar" not in st.session_state:
- st.session_state.user_avatar = DEFAULT_USER_AVATAR
-if "bot_avatar" not in st.session_state:
- st.session_state.bot_avatar = DEFAULT_BOT_AVATAR
-
-# Define avatar images
-USER_AVATAR = "👤" # Default user emoji
-BOT_AVATAR = "🤖" # Default bot emoji
-
-
-class StreamlitProgressReporter(PrintProgressReporter):
- def __init__(self, placeholder):
- super().__init__("")
- self.placeholder = placeholder
-
- def success(self, message: str):
- self.placeholder.success(message)
-
-
-def render_chat_tab():
- """Render the Chat tab content."""
- format_message_history()
-
- # Chat input
- if prompt := st.chat_input("Enter your query..."):
- # Add user message to history with timestamp
- st.session_state.messages.append(
- {
- "role": "user",
- "content": prompt,
- "timestamp": pd.Timestamp.now().strftime("%H:%M"),
- }
- )
-
- # Process query
- with st.spinner("Processing your query..."):
- response_placeholder = st.empty()
- try:
- if st.session_state.search_type == "global":
- response, context = run_global_search(
- config_filepath=st.session_state.config_filepath,
- data_dir=st.session_state.data_dir,
- root_dir=st.session_state.root_dir,
- community_level=st.session_state.community_level,
- response_type=st.session_state.response_type,
- streaming=st.session_state.streaming,
- query=prompt,
- progress_placeholder=response_placeholder,
- )
- elif st.session_state.search_type == "drift":
- response, context = run_drift_search(
- config_filepath=st.session_state.config_filepath,
- data_dir=st.session_state.data_dir,
- root_dir=st.session_state.root_dir,
- community_level=st.session_state.community_level,
- response_type=st.session_state.response_type,
- streaming=st.session_state.streaming,
- query=prompt,
- progress_placeholder=response_placeholder,
- )
- else:
- response, context = run_local_search(
- config_filepath=st.session_state.config_filepath,
- data_dir=st.session_state.data_dir,
- root_dir=st.session_state.root_dir,
- community_level=st.session_state.community_level,
- response_type=st.session_state.response_type,
- streaming=st.session_state.streaming,
- query=prompt,
- progress_placeholder=response_placeholder,
- )
-
- # Clear the placeholder before adding the final response
- response_placeholder.empty()
-
- # Add assistant response to history with timestamp
- st.session_state.messages.append(
- {
- "role": "assistant",
- "content": response,
- "timestamp": pd.Timestamp.now().strftime("%H:%M"),
- }
- )
-
- # Show context in expander
- with st.expander("View Search Context"):
- st.json(context)
-
- except Exception as e:
- error_message = f"Error processing query: {str(e)}"
- st.session_state.messages.append(
- {
- "role": "assistant",
- "content": error_message,
- "timestamp": pd.Timestamp.now().strftime("%H:%M"),
- }
- )
-
-
- st.rerun()
-
-
-def display_message(msg: str, is_user: bool = False, timestamp: str = "") -> None:
- """Display a chat message with avatar and consistent formatting."""
- role = "user" if is_user else "assistant"
- message_class = "user-message" if is_user else "assistant-message"
- avatar = st.session_state.user_avatar if is_user else st.session_state.bot_avatar
-
- message_container = f"""
-
- """
- st.markdown(message_container, unsafe_allow_html=True)
-
-
-def format_message_history() -> None:
- """Display all messages in the chat history with consistent formatting."""
- st.markdown('', unsafe_allow_html=True)
- for message in st.session_state.messages:
- timestamp = message.get("timestamp", "")
- display_message(
- msg=message["content"],
- is_user=(message["role"] == "user"),
- timestamp=timestamp,
- )
- st.markdown("
", unsafe_allow_html=True)
-
-
-@st.cache_resource
-def load_css():
- with open("styles.css", "r") as f:
- return f.read()
-
-
-def initialize_session_state():
- """Initialize session state variables if they don't exist."""
- if "messages" not in st.session_state:
- st.session_state.messages = []
- if "response_placeholder" not in st.session_state:
- st.session_state.response_placeholder = None
- if "config_filepath" not in st.session_state:
- st.session_state.config_filepath = None
- if "data_dir" not in st.session_state:
- st.session_state.data_dir = None
- if "root_dir" not in st.session_state:
- st.session_state.root_dir = "."
- if "community_level" not in st.session_state:
- st.session_state.community_level = 2
- if "response_type" not in st.session_state:
- st.session_state.response_type = "concise"
- if "search_type" not in st.session_state:
- st.session_state.search_type = "global"
- if "streaming" not in st.session_state:
- st.session_state.streaming = True
- if "authenticated" not in st.session_state:
- st.session_state.authenticated = False
-
-
-def main():
- initialize_session_state()
-
- # Authentication check
- if not st.session_state.authenticated:
- if auth.check_credentials():
- st.session_state.authenticated = True
- st.rerun() # Rerun to reflect the authentication state
- else:
- st.stop() # Stop further execution if authentication fails
-
- # If authenticated, proceed with the main app
- if st.session_state.authenticated:
- # Main application content
- st.title("PWC Home Assigment #1, Graphrag")
-
- css = load_css()
- st.markdown(f"", unsafe_allow_html=True)
-
- # Sidebar configuration
- with st.sidebar:
- # Display logos side by side at the top of the sidebar
- col1, col2 = st.columns(2)
- with col1:
- st.markdown(
- '',
- unsafe_allow_html=True,
- )
- with col2:
- st.markdown(
- '',
- unsafe_allow_html=True,
- )
-
- st.header("Configuration")
- st.session_state.community_level = st.number_input(
- "Community Level",
- min_value=0,
- max_value=10,
- value=st.session_state.community_level,
- help="Controls the granularity of the search...",
- )
-
- # Only show response type for global and local search
- if st.session_state.search_type != "drift":
- st.session_state.response_type = st.selectbox(
- "Response Type",
- options=["concise", "detailed"],
- index=0 if st.session_state.response_type == "concise" else 1,
- help="Style of response generation",
- )
-
- st.session_state.search_type = st.selectbox(
- "Search Type",
- options=["global", "local", "drift"],
- index=(
- 0
- if st.session_state.search_type == "global"
- else 1 if st.session_state.search_type == "local" else 2
- ),
- help="""Search Types:
- - Local Search: "Focuses on finding specific information by searching through direct connections in the knowledge graph. Best for precise, fact-based queries."
- - Global Search: "Analyzes the entire document collection at a high level using community summaries. Best for understanding broad themes and general policies."
- - DRIFT Search: "Combines local and global search capabilities, dynamically exploring connections while gathering detailed information. Best for complex queries requiring both specific details and broader context."
- """,
- )
-
- # Show streaming option only for supported search types
- if st.session_state.search_type != "drift":
- st.session_state.streaming = st.checkbox(
- "Enable Streaming",
- value=st.session_state.streaming,
- help="Stream response tokens as they're generated",
- )
- else:
- st.session_state.streaming = False
- st.info("Streaming is not available for DRIFT search")
-
- # logout button
- if st.button("Logout"):
- st.session_state.clear() # Clear all session state data
- initialize_session_state() # Reinitialize the session state
- st.query_params = {"restart": "true"} # Refresh the UI
- st.rerun()
-
- # Create tabs
- tab1, tab2 = st.tabs(["Assignment Documentation", "Chat"])
-
- # readme tab content
- with tab1:
- render_wiki_tab()
-
- # Chat tab content
- with tab2:
- render_chat_tab()
-
- st.sidebar.markdown(
- """
-
- """,
- unsafe_allow_html=True,
- )
-
-
-if __name__ == "__main__":
- main()
-=======
-import streamlit as st
-import asyncio
-import sys
-from pathlib import Path
-import base64
-import pandas as pd
-from typing import Literal, Tuple, Optional
-from wiki import render_wiki_tab
-from search_handlers import run_global_search, run_local_search, run_drift_search
-import auth
-
-
-import graphrag.api as api
-from graphrag.config import GraphRagConfig, load_config, resolve_paths
-from graphrag.index.create_pipeline_config import create_pipeline_config
-from graphrag.logging import PrintProgressReporter
-from graphrag.utils.storage import _create_storage, _load_table_from_storage
-
-
-st.set_page_config(page_title="GraphRAG Chat Interface", page_icon="🔍", layout="wide")
-
-# Define default avatars at the module level
-DEFAULT_USER_AVATAR = "👤"
-DEFAULT_BOT_AVATAR = "🤖"
-
-# Initialize session state for avatars
-if "user_avatar" not in st.session_state:
- st.session_state.user_avatar = DEFAULT_USER_AVATAR
-if "bot_avatar" not in st.session_state:
- st.session_state.bot_avatar = DEFAULT_BOT_AVATAR
-
-# Define avatar images
-USER_AVATAR = "👤" # Default user emoji
-BOT_AVATAR = "🤖" # Default bot emoji
-
-
-class StreamlitProgressReporter(PrintProgressReporter):
- def __init__(self, placeholder):
- super().__init__("")
- self.placeholder = placeholder
-
- def success(self, message: str):
- self.placeholder.success(message)
-
-
-def render_chat_tab():
- """Render the Chat tab content."""
- format_message_history()
-
- # Chat input
- if prompt := st.chat_input("Enter your query..."):
- # Add user message to history with timestamp
- st.session_state.messages.append(
- {
- "role": "user",
- "content": prompt,
- "timestamp": pd.Timestamp.now().strftime("%H:%M"),
- }
- )
-
- # Process query
- with st.spinner("Processing your query..."):
- response_placeholder = st.empty()
- try:
- if st.session_state.search_type == "global":
- response, context = run_global_search(
- config_filepath=st.session_state.config_filepath,
- data_dir=st.session_state.data_dir,
- root_dir=st.session_state.root_dir,
- community_level=st.session_state.community_level,
- response_type=st.session_state.response_type,
- streaming=st.session_state.streaming,
- query=prompt,
- progress_placeholder=response_placeholder,
- )
- elif st.session_state.search_type == "drift":
- response, context = run_drift_search(
- config_filepath=st.session_state.config_filepath,
- data_dir=st.session_state.data_dir,
- root_dir=st.session_state.root_dir,
- community_level=st.session_state.community_level,
- response_type=st.session_state.response_type,
- streaming=st.session_state.streaming,
- query=prompt,
- progress_placeholder=response_placeholder,
- )
- else:
- response, context = run_local_search(
- config_filepath=st.session_state.config_filepath,
- data_dir=st.session_state.data_dir,
- root_dir=st.session_state.root_dir,
- community_level=st.session_state.community_level,
- response_type=st.session_state.response_type,
- streaming=st.session_state.streaming,
- query=prompt,
- progress_placeholder=response_placeholder,
- )
-
- # Clear the placeholder before adding the final response
- response_placeholder.empty()
-
- # Add assistant response to history with timestamp
- st.session_state.messages.append(
- {
- "role": "assistant",
- "content": response,
- "timestamp": pd.Timestamp.now().strftime("%H:%M"),
- }
- )
-
- # Show context in expander
- with st.expander("View Search Context"):
- st.json(context)
-
- except Exception as e:
- error_message = f"Error processing query: {str(e)}"
- st.session_state.messages.append(
- {
- "role": "assistant",
- "content": error_message,
- "timestamp": pd.Timestamp.now().strftime("%H:%M"),
- }
- )
-
-
- st.rerun()
-
-
-def display_message(msg: str, is_user: bool = False, timestamp: str = "") -> None:
- """Display a chat message with avatar and consistent formatting."""
- role = "user" if is_user else "assistant"
- message_class = "user-message" if is_user else "assistant-message"
- avatar = st.session_state.user_avatar if is_user else st.session_state.bot_avatar
-
- message_container = f"""
-
- """
- st.markdown(message_container, unsafe_allow_html=True)
-
-
-def format_message_history() -> None:
- """Display all messages in the chat history with consistent formatting."""
- st.markdown('', unsafe_allow_html=True)
- for message in st.session_state.messages:
- timestamp = message.get("timestamp", "")
- display_message(
- msg=message["content"],
- is_user=(message["role"] == "user"),
- timestamp=timestamp,
- )
- st.markdown("
", unsafe_allow_html=True)
-
-
-@st.cache_resource
-def load_css():
- with open("styles.css", "r") as f:
- return f.read()
-
-
-def initialize_session_state():
- """Initialize session state variables if they don't exist."""
- if "messages" not in st.session_state:
- st.session_state.messages = []
- if "response_placeholder" not in st.session_state:
- st.session_state.response_placeholder = None
- if "config_filepath" not in st.session_state:
- st.session_state.config_filepath = None
- if "data_dir" not in st.session_state:
- st.session_state.data_dir = None
- if "root_dir" not in st.session_state:
- st.session_state.root_dir = "."
- if "community_level" not in st.session_state:
- st.session_state.community_level = 2
- if "response_type" not in st.session_state:
- st.session_state.response_type = "concise"
- if "search_type" not in st.session_state:
- st.session_state.search_type = "global"
- if "streaming" not in st.session_state:
- st.session_state.streaming = True
- if "authenticated" not in st.session_state:
- st.session_state.authenticated = False
-
-
-def main():
- initialize_session_state()
-
- # Authentication check
- if not st.session_state.authenticated:
- if auth.check_credentials():
- st.session_state.authenticated = True
- st.rerun() # Rerun to reflect the authentication state
- else:
- st.stop() # Stop further execution if authentication fails
-
- # If authenticated, proceed with the main app
- if st.session_state.authenticated:
- # Main application content
- st.title("PWC Home Assigment #1, Graphrag")
-
- css = load_css()
- st.markdown(f"", unsafe_allow_html=True)
-
- # Sidebar configuration
- with st.sidebar:
- # Display logos side by side at the top of the sidebar
- col1, col2 = st.columns(2)
- with col1:
- st.markdown(
- '
