Update app.py

app.py

@@ -8,7 +8,6 @@ import os
 import gradio as gr
 from smolagents import CodeAgent, HfApiModel, tool
 from typing import Dict, List, Optional, Tuple, Union
-import json
 
 # Advanced Research Tools
 @tool
@@ -40,19 +39,10 @@ def web_search(query: str, max_results: int = 5) -> str:
 
 @tool
 def analyze_content(text: str, analysis_type: str = "general") -> str:
-    """Analyzes content for various aspects like key points, themes, or citations
-    
-    Args:
-        text: The content to analyze
-        analysis_type: Type of analysis ("general", "academic", "citations")
-    
-    Returns:
-        str: Analysis results in a structured format
-    """
+    """Analyzes content for various aspects like key points, themes, or citations"""
     points = []
     
     if "academic" in analysis_type.lower():
-        # Academic analysis focusing on scholarly aspects
         points.extend([
             "Key Arguments:",
             "- Main thesis and supporting evidence",
@@ -63,7 +53,6 @@ def analyze_content(text: str, analysis_type: str = "general") -> str:
             "- Research foundations",
         ])
     elif "citations" in analysis_type.lower():
-        # Citation analysis
         points.extend([
             "Citation Analysis:",
             "- Author credentials",
@@ -74,7 +63,6 @@ def analyze_content(text: str, analysis_type: str = "general") -> str:
             "- Secondary references",
         ])
     else:
-        # General content analysis
         points.extend([
             "Content Overview:",
             "- Main topics and themes",
@@ -137,7 +125,8 @@ class ResearchSystem:
                     chat = gr.Chatbot(
                         label="Research Process",
                         height=600,
-                        show_label=True
+                        show_label=True,
+                        type="messages"  # Use new message format
                     )
                     
                 with gr.Column(scale=1):
@@ -181,11 +170,13 @@ class ResearchSystem:
                 outputs=[chat]
             )
             
+            # Clear chat using the proper method
+            def clear_chat():
+                return None
+            
             clear_btn.click(
-                lambda: None,
-                inputs=[],
-                outputs=[chat],
-                _js="() => []"  # Clear chat
+                clear_chat,
+                outputs=[chat]
             )
             
             # Help information
@@ -206,7 +197,7 @@ class ResearchSystem:
     def process_query(self, 
                      query: str, 
                      depth: int = 3, 
-                     num_sources: int = 5) -> List[List[str]]:
+                     num_sources: int = 5) -> List[Dict[str, str]]:
         """
         Processes a research query with enhanced capabilities
         
@@ -216,16 +207,16 @@ class ResearchSystem:
             num_sources: Number of sources to include (1-10)
             
         Returns:
-            List of message pairs for the chatbot interface
+            List of message dictionaries for the chatbot interface
         """
         try:
             # Check cache first
             if query in self.cache and self.config["cache_results"]:
                 raw_research, formatted = self.cache[query]
                 return [
-                    [query, None],
-                    ["Researcher (Cached)", raw_research],
-                    ["Formatter (Cached)", formatted]
+                    {"role": "user", "content": query},
+                    {"role": "assistant", "content": f"📚 Research Findings:\n{raw_research}"},
+                    {"role": "assistant", "content": f"📝 MLA Formatted:\n{formatted}"}
                 ]
             
             # Configure research parameters
@@ -256,11 +247,11 @@ class ResearchSystem:
             if self.config["cache_results"]:
                 self.cache[query] = (raw_research, formatted)
             
-            # Return results in chatbot format
+            # Return results in new message format
             return [
-                [query, None],
-                ["📚 Research Findings", raw_research],
-                ["📝 MLA Formatted", formatted]
+                {"role": "user", "content": query},
+                {"role": "assistant", "content": f"📚 Research Findings:\n{raw_research}"},
+                {"role": "assistant", "content": f"📝 MLA Formatted:\n{formatted}"}
             ]
             
         except Exception as e:
@@ -268,7 +259,7 @@ class ResearchSystem:
                 f"Error during research process: {str(e)}\n"
                 "Please try again or refine your query."
             )
-            return [[None, error_msg]]
+            return [{"role": "assistant", "content": error_msg}]
 
 if __name__ == "__main__":
     system = ResearchSystem()