Update app.py

app.py

@@ -6,19 +6,11 @@ from tools.perplexity_tools import (
     explain_concept,
 )
 from tools.general_tools import initialize_perplexity
-import time
 
 # Initialize Perplexity without API key and model
 perplexity_api_key = None
 perplexity_model = "sonar-pro"  # Default model
 
-# Function to simulate streaming
-def simulate_streaming(text: str, delay: float = 0.05):
-    """Simulates streaming by yielding text incrementally."""
-    for char in text:
-        time.sleep(delay)
-        yield char
-
 # Function to handle chat UI interactions
 def chat_ui_interaction(query: str, chat_history: list):
     """Handles user queries in the chat UI, showing the agent's thought process."""
@@ -26,52 +18,50 @@ def chat_ui_interaction(query: str, chat_history: list):
     # Add user message to chat history
     chat_history.append(("user", query))
     
-    # Let the agent decide which tools to invoke
+    # Let the agent decide which tool to invoke
     thought_process = []
     try:
         # Simulate the agent's thought process
         thought_process.append("🤔 Analyzing your query...")
-        yield chat_history + [("assistant", "🤔 Analyzing your query...")]
         
         # Example: Decide which tool to use based on the query
         if "explain" in query.lower():
             thought_process.append("🔍 Using the 'explain_concept' tool...")
-            yield chat_history + [("assistant", "🔍 Using the 'explain_concept' tool...")]
             result = explain_concept(query, perplexity_api_key, perplexity_model)
         elif "summarize" in query.lower():
             thought_process.append("📄 Using the 'summarize_paper' tool...")
-            yield chat_history + [("assistant", "📄 Using the 'summarize_paper' tool...")]
             result = summarize_paper(query, perplexity_api_key, perplexity_model)
         elif "citation" in query.lower():
             thought_process.append("📚 Using the 'get_citation' tool...")
-            yield chat_history + [("assistant", "📚 Using the 'get_citation' tool...")]
             result = get_citation(query, perplexity_api_key, perplexity_model)
         else:
             thought_process.append("🔎 Using the 'get_ai_research_papers' tool...")
-            yield chat_history + [("assistant", "🔎 Using the 'get_ai_research_papers' tool...")]
             result = get_ai_research_papers(query, perplexity_api_key, perplexity_model)
         
-        # Simulate streaming the final result
-        for chunk in simulate_streaming(result):
-            chat_history.append(("assistant", chunk))
-            yield chat_history
+        # Add thought process to chat history
+        for step in thought_process:
+            chat_history.append(("assistant", step))
+        
+        # Add final result to chat history
+        chat_history.append(("assistant", result))
     except Exception as e:
         chat_history.append(("assistant", f"Error: {str(e)}"))
-        yield chat_history
+    
+    return chat_history
 
 # Enhanced UI with toggle between current UI and chat UI
 def create_ui():
     with gr.Blocks() as demo:
         # Toggle button to switch between UIs
         toggle_button = gr.Radio(
-            choices=["Current UI", "Chat UI"],
+            choices=["Specific UI", "Agentic UI"],
             label="Select UI Mode",
-            value="Current UI"
+            value="Specific UI"
         )
         
-        # Current UI components
-        with gr.Column(visible=True) as current_ui:
-            gr.Markdown("# AI Research Assistant")
+        # Specific UI components
+        with gr.Column(visible=True) as specific_ui:
+            gr.Markdown("# AI Research Assistant (Specific Mode)")
             gr.Markdown("Your AI-powered research companion")
             
             with gr.Row():
@@ -142,22 +132,18 @@ def create_ui():
                 outputs=explanation_output
             )
         
-        # Chat UI components
-        with gr.Column(visible=False) as chat_ui:
-            gr.Markdown("# AI Research Assistant (Chat Mode)")
+        # Agentic UI (Chat UI) components
+        with gr.Column(visible=False) as agentic_ui:
+            gr.Markdown("# AI Research Assistant (Agentic Mode)")
             
-            # Chat layout
+            # Chat interface
             with gr.Row():
-                # Left: Threads (conversation history)
+                # Left column for threads (optional)
                 with gr.Column(scale=1):
-                    threads = gr.Dataframe(
-                        headers=["Thread ID", "Title"],
-                        value=[["1", "AI Research"], ["2", "Paper Summaries"]],
-                        interactive=False,
-                        label="Threads"
-                    )
+                    gr.Markdown("### Threads")
+                    threads = gr.Textbox(label="Threads", interactive=False)
                 
-                # Center: Messages (chat history)
+                # Center column for chat messages
                 with gr.Column(scale=3):
                     chatbot = gr.Chatbot(label="Chat History")
                     chat_input = gr.Textbox(label="Your Message", placeholder="Type your message here...")
@@ -172,7 +158,7 @@ def create_ui():
         
         # Toggle between UIs
         def toggle_ui(mode):
-            if mode == "Current UI":
+            if mode == "Specific UI":
                 return [gr.Column.update(visible=True), gr.Column.update(visible=False)]
             else:
                 return [gr.Column.update(visible=False), gr.Column.update(visible=True)]
@@ -180,7 +166,7 @@ def create_ui():
         toggle_button.change(
             fn=toggle_ui,
             inputs=toggle_button,
-            outputs=[current_ui, chat_ui]
+            outputs=[specific_ui, agentic_ui]
         )
     
     return demo