Update app.py

app.py

@@ -1,178 +1,6 @@
-
-import gradio as gr
-from tools.perplexity_tools import (
-    get_ai_research_papers,
-    summarize_paper,
-    get_citation,
-    explain_concept,
-)
-from tools.general_tools import initialize_perplexity
-
-# Initialize Perplexity without API key and model
-perplexity_api_key = None
-perplexity_model = "sonar-pro"  # Default model
-
-# 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."""
-    global perplexity_api_key, perplexity_model
-    # Add user message to chat history
-    chat_history.append(("user", query))
-    
-    # Let the agent decide which tool to invoke
-    thought_process = []
-    try:
-        # Simulate the agent's thought process
-        thought_process.append("🤔 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...")
-            result = explain_concept(query, perplexity_api_key, perplexity_model)
-        elif "summarize" in query.lower():
-            thought_process.append("📄 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...")
-            result = get_citation(query, perplexity_api_key, perplexity_model)
-        else:
-            thought_process.append("🔎 Using the 'get_ai_research_papers' tool...")
-            result = get_ai_research_papers(query, perplexity_api_key, perplexity_model)
-        
-        # 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)}"))
-    
-    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=["Specific UI", "Agentic UI"],
-            label="Select UI Mode",
-            value="Specific UI"
-        )
-        
-        # 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():
-                gr.Markdown("## Setup Required")
-                gr.Markdown("1. Enter Perplexity API key and select a model")
-                input_api_key = gr.Textbox(label="Enter your Perplexity API Key:")
-                model_dropdown = gr.Dropdown(
-                    label="Select Perplexity Model",
-                    choices=[
-                        "sonar-reasoning-pro",
-                        "sonar-reasoning",
-                        "sonar-pro",
-                        "sonar"
-                    ],
-                    value="sonar-pro"  # Default value
-                )
-                initialize_button = gr.Button("Initialize Perplexity API")
-            
-            initialization_status = gr.Textbox(label="Initialization Status")
-            initialize_button.click(
-                fn=initialize_perplexity,
-                inputs=[input_api_key, model_dropdown],
-                outputs=initialization_status
-            )
-            
-            with gr.Row():
-                gr.Markdown("## Research Tools")
-                input_query = gr.Textbox(label="Search query for AI papers:")
-                search_button = gr.Button("Search Papers")
-            
-            search_results = gr.Textbox(label="Search Results")
-            search_button.click(
-                fn=get_ai_research_papers,
-                inputs=[input_query, input_api_key, model_dropdown],
-                outputs=search_results
-            )
-            
-            with gr.Row():
-                input_paper_title = gr.Textbox(label="Paper title to summarize:")
-                summarize_button = gr.Button("Summarize Paper")
-            
-            paper_summary = gr.Textbox(label="Paper Summary")
-            summarize_button.click(
-                fn=summarize_paper,
-                inputs=[input_paper_title, input_api_key, model_dropdown],
-                outputs=paper_summary
-            )
-            
-            with gr.Row():
-                input_citation_title = gr.Textbox(label="Paper title to cite:")
-                citation_button = gr.Button("Generate Citation")
-            
-            citation_output = gr.Textbox(label="Citation")
-            citation_button.click(
-                fn=get_citation,
-                inputs=[input_citation_title, input_api_key, model_dropdown],
-                outputs=citation_output
-            )
-            
-            with gr.Row():
-                input_concept = gr.Textbox(label="Concept to explain:")
-                explain_button = gr.Button("Explain Concept")
-            
-            explanation_output = gr.Textbox(label="Explanation")
-            explain_button.click(
-                fn=explain_concept,
-                inputs=[input_concept, input_api_key, model_dropdown],
-                outputs=explanation_output
-            )
-        
-        # Agentic UI (Chat UI) components
-        with gr.Column(visible=False) as agentic_ui:
-            gr.Markdown("# AI Research Assistant (Agentic Mode)")
-            
-            # Chat interface
-            with gr.Row():
-                # Left column for threads (optional)
-                with gr.Column(scale=1):
-                    gr.Markdown("### Threads")
-                    threads = gr.Textbox(label="Threads", interactive=False)
-                
-                # 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...")
-                    chat_button = gr.Button("Send")
-            
-            # Handle chat interactions
-            chat_button.click(
-                fn=chat_ui_interaction,
-                inputs=[chat_input, chatbot],
-                outputs=chatbot
-            )
-        
-        # Toggle between UIs
-        def toggle_ui(mode):
-            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)]
-        
-        toggle_button.change(
-            fn=toggle_ui,
-            inputs=toggle_button,
-            outputs=[specific_ui, agentic_ui]
-        )
-    
-    return demo
+from main_ui import create_main_ui
 
 # Launch the enhanced UI
 if __name__ == "__main__":
-    demo = create_ui()
+    demo = create_main_ui()
     demo.launch()