Update app.py

app.py

@@ -18,13 +18,6 @@ CHATBOT_SYSTEM_PROMPT = """You are a helpful AI assistant that analyzes Hugging
 Your task is to help users understand repositories, extract key information, and provide insights.
 Be concise, clear, and focus on the most important aspects of each repository."""
 
-class AppState:
-    """Simple state management for the application."""
-    def __init__(self):
-        self.repo_ids: List[str] = []
-        self.current_repo_idx: int = 0
-        self.chat_history: List[Dict[str, str]] = []
-
 def read_csv_as_text(filename: str) -> pd.DataFrame:
     """Read CSV file and return as DataFrame."""
     try:
@@ -78,10 +71,9 @@ def analyze_repo(repo_id: str) -> Tuple[str, str]:
         logger.error(f"Error analyzing repo {repo_id}: {e}")
         return f"Error analyzing {repo_id}", f"Error: {str(e)}"
 
-def chat_with_user(message: str, history: List[Dict[str, str]], system_prompt: str) -> str:
+def chat_with_user(message: str, history: List[Dict[str, str]]) -> str:
     """Simple chat response."""
     try:
-        # For now, return a simple response
         return f"I understand you're asking about: {message}. How can I help you analyze this repository?"
     except Exception as e:
         logger.error(f"Error in chat: {e}")
@@ -89,8 +81,6 @@ def chat_with_user(message: str, history: List[Dict[str, str]], system_prompt: s
 
 def create_ui() -> gr.Blocks:
     """Create a simplified Gradio interface."""
-    state = gr.State(AppState())
-    
     with gr.Blocks(title="Hugging Face Repo Analyzer", theme=gr.themes.Soft()) as app:
         gr.Markdown("# Hugging Face Repository Analyzer")
         
@@ -117,13 +107,17 @@ def create_ui() -> gr.Blocks:
                 summary_output = gr.Textbox(label="Analysis Summary", lines=5)
                 
                 # Chat Section
-                chatbot = gr.Chatbot(label="Chat with Assistant", height=400)
+                chatbot = gr.Chatbot(
+                    label="Chat with Assistant",
+                    height=400,
+                    type="messages"
+                )
                 msg = gr.Textbox(label="Message", placeholder="Ask about the repository...")
                 with gr.Row():
                     send_btn = gr.Button("Send", variant="primary")
                     clear_btn = gr.Button("Clear Chat", variant="secondary")
         
-        def process_input(text: str, state: AppState) -> Tuple[pd.DataFrame, str, str, str]:
+        def process_input(text: str) -> Tuple[pd.DataFrame, str, str, str]:
             """Process input and return results."""
             try:
                 # Check if input is keywords or repo IDs
@@ -143,9 +137,7 @@ def create_ui() -> gr.Blocks:
                 if not repo_ids:
                     return pd.DataFrame(), "No repositories found", "", ""
                 
-                # Update state and CSV
-                state.repo_ids = repo_ids
-                state.current_repo_idx = 0
+                # Update CSV
                 write_repos_to_csv(repo_ids)
                 
                 # Get first repo analysis
@@ -157,12 +149,12 @@ def create_ui() -> gr.Blocks:
                 logger.error(f"Error processing input: {e}")
                 return pd.DataFrame(), f"Error: {str(e)}", "", ""
         
-        def send_message(message: str, history: List[Dict[str, str]], state: AppState) -> Tuple[List[Dict[str, str]], str]:
+        def send_message(message: str, history: List[Dict[str, str]]) -> Tuple[List[Dict[str, str]], str]:
             """Send message to chat."""
             if not message:
                 return history, ""
             history.append({"role": "user", "content": message})
-            response = chat_with_user(message, history, CHATBOT_SYSTEM_PROMPT)
+            response = chat_with_user(message, history)
             history.append({"role": "assistant", "content": response})
             return history, ""
         
@@ -173,13 +165,13 @@ def create_ui() -> gr.Blocks:
         # Event handlers
         submit_btn.click(
             fn=process_input,
-            inputs=[repo_input, state],
+            inputs=[repo_input],
             outputs=[df_output, status, content_output, summary_output]
         )
         
         send_btn.click(
             fn=send_message,
-            inputs=[msg, chatbot, state],
+            inputs=[msg, chatbot],
             outputs=[chatbot, msg]
         )