fixing everything

Dockerfile

@@ -1,23 +1,18 @@
-# Use official Python 3.9 image
 FROM python:3.9
-
-# Create and use a non-root user
 RUN useradd -m -u 1000 user
 USER user
-
-# Define environment variables
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH \
-    PYTHONPATH=/home/user/app 
-
-# Set working directory
+    PATH=/home/user/.local/bin:$PATH
 WORKDIR $HOME/app
 
-# Copy files with proper ownership
-COPY --chown=user . $HOME/app
+# Copy requirements first for better caching
+COPY --chown=user requirements.txt .
+RUN pip install -r requirements.txt
+
+# Copy the rest of the application
+COPY --chown=user . .
 
-# Install dependencies
-RUN pip install --no-cache-dir -r requirements.txt
+ENV PYTHONUNBUFFERED=1
+EXPOSE 7860
 
-# Run the application
-CMD ["chainlit", "run", "app.py", "--port", "7860"]
+CMD ["chainlit", "run", "app.py", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 import os
 import chainlit as cl
 import pandas as pd
-from typing import List, Dict, Any, TypedDict, Callable, Annotated, Literal, Optional, Union, Tuple
+from typing import List, Dict, Any, TypedDict, Callable, Annotated, Literal, Optional, Union, Tuple, TypeVar
 from qdrant_client import QdrantClient
 from qdrant_client.models import Distance, VectorParams, PointStruct
 from langchain_core.messages import HumanMessage, AIMessage, SystemMessage
@@ -20,6 +20,8 @@ from langchain_core.runnables import RunnableConfig
 from langchain_core.runnables.utils import Output
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
+from dataclasses import dataclass, field
+from state import FounderAnalysisState
 
 # Load environment variables
 load_dotenv()
@@ -40,17 +42,7 @@ SIMILARITY_THRESHOLD = 0.75
 DEFAULT_TIMEOUT = 60  # Default timeout in seconds
 API_RATE_LIMIT_DELAY = 1  # Delay between API calls in seconds
 
-# Define state structure for LangGraph
-class FounderAnalysisState(TypedDict):
-    query: str
-    query_type: str  # "search" or "filter"
-    filter_key: str
-    filter_value: str
-    retrieved_profiles: List[Dict[str, Any]]
-    web_search_results: List[Dict[str, Any]]
-    analysis_results: List[Dict[str, Any]]
-    final_response: Dict[str, Any]
-    error: str
+StateType = TypeVar("StateType", bound=Dict[str, Any])
 
 # Decorator for adding timeouts to async functions
 def async_timeout(timeout_seconds=DEFAULT_TIMEOUT):
@@ -142,8 +134,8 @@ class FounderAnalysisSystem:
 
     def _create_workflow(self) -> StateGraph:
         """Create the LangGraph workflow for founder analysis."""
-        # Create a new graph
-        workflow = StateGraph(FounderAnalysisState)
+        # Use a simple dict type for the state graph
+        workflow = StateGraph(dict)
         
         # Add nodes to the graph
         workflow.add_node("process_query", self.process_query)
@@ -190,6 +182,22 @@ class FounderAnalysisSystem:
     @async_timeout(30)  # 30 second timeout for query processing
     async def process_query(self, state: FounderAnalysisState) -> FounderAnalysisState:
         """Process the user query and determine the query type."""
+        # Initialize state if needed
+        if not isinstance(state, dict):
+            state = {}
+        
+        state.update({
+            "query": state.get("query", ""),
+            "query_type": "",
+            "filter_key": "",
+            "filter_value": "",
+            "retrieved_profiles": [],
+            "web_search_results": [],
+            "analysis_results": [],
+            "final_response": {},
+            "error": ""
+        })
+        
         query = state["query"]
         
         # Log the processing step
@@ -616,18 +624,18 @@ class FounderAnalysisSystem:
         # Reset progress message
         self.progress_message = None
         
-        # Initialize the state
-        state = FounderAnalysisState(
-            query=query,
-            query_type="",
-            filter_key="",
-            filter_value="",
-            retrieved_profiles=[],
-            web_search_results=[],
-            analysis_results=[],
-            final_response={},
-            error=""
-        )
+        # Initialize the state as a simple dictionary
+        state = {
+            "query": query,
+            "query_type": "",
+            "filter_key": "",
+            "filter_value": "",
+            "retrieved_profiles": [],
+            "web_search_results": [],
+            "analysis_results": [],
+            "final_response": {},
+            "error": ""
+        }
         
         try:
             # Manually execute the workflow nodes in sequence

state.py

@@ -0,0 +1,12 @@
+from typing import TypedDict, List, Dict, Any
+
+class FounderAnalysisState(TypedDict, total=False):
+    query: str
+    query_type: str
+    filter_key: str
+    filter_value: str
+    retrieved_profiles: List[Dict[str, Any]]
+    web_search_results: List[Dict[str, Any]]
+    analysis_results: List[Dict[str, Any]]
+    final_response: Dict[str, Any]
+    error: str