updated api respondes v2

mcp_server.py

@@ -609,6 +609,85 @@ Sample analysis content..."""
             
         return interface
 
+# Create a FastAPI app for direct API access
+app = gr.mount_gradio_app(
+    gr.routes.App(app=gr.Blocks()),
+    create_interface(),
+    path="/"
+)
+
+# Add a direct API endpoint for research
+@app.post("/api/research")
+async def api_research(request: dict):
+    try:
+        query = request.get('query', '')
+        if not query:
+            return {"error": "Query parameter is required"}, 400
+            
+        # Initialize with default values or from environment variables
+        api_type = os.getenv("DEFAULT_API_TYPE", "Gemini")
+        gemini_key = os.getenv("GEMINI_API_KEY")
+        gemini_model = os.getenv("GEMINI_MODEL", "gemini-2.0-flash")
+        tavily_key = os.getenv("TAVILY_API_KEY")
+        openrouter_key = os.getenv("OPENROUTER_API_KEY")
+        openrouter_model = os.getenv("OPENROUTER_MODEL", "anthropic/claude-3-opus:beta")
+        
+        # Override with request values if provided
+        if 'api_type' in request:
+            api_type = request['api_type']
+        if 'gemini_key' in request:
+            gemini_key = request['gemini_key']
+        if 'gemini_model' in request:
+            gemini_model = request['gemini_model']
+        if 'tavily_key' in request:
+            tavily_key = request['tavily_key']
+        if 'openrouter_key' in request:
+            openrouter_key = request['openrouter_key']
+        if 'openrouter_model' in request:
+            openrouter_model = request['openrouter_model']
+            
+        # Validate required API keys
+        if not tavily_key:
+            return {"error": "Tavily API key is required"}, 400
+            
+        if api_type == "Gemini" and not gemini_key:
+            return {"error": "Gemini API key is required when using Gemini"}, 400
+            
+        if api_type == "OpenRouter" and not openrouter_key:
+            return {"error": "OpenRouter API key is required when using OpenRouter"}, 400
+        
+        # Initialize the system
+        system = MultiAgentSystem(
+            use_gemini=(api_type == "Gemini"),
+            gemini_api_key=gemini_key,
+            gemini_model=gemini_model,
+            tavily_api_key=tavily_key,
+            openrouter_api_key=openrouter_key if api_type == "OpenRouter" else None,
+            openrouter_model=openrouter_model if api_type == "OpenRouter" else None
+        )
+        
+        # Process the query
+        markdown_text = system.process_query(query)
+        
+        # Generate file paths
+        md_file_path = save_markdown_report(markdown_text)
+        html_file_path = convert_to_html(markdown_text)
+        
+        # Return the results
+        return {
+            "query": query,
+            "result": markdown_text,
+            "downloads": {
+                "markdown": md_file_path,
+                "html": html_file_path
+            },
+            "status": "success"
+        }
+        
+    except Exception as e:
+        server_logger.error(f"API research failed: {str(e)}", exc_info=True)
+        return {"error": f"Research failed: {str(e)}"}, 500
+
 if __name__ == "__main__":
     try:
         # Configure event loop policy for Windows
@@ -618,15 +697,15 @@ if __name__ == "__main__":
             if sys.version_info[0] == 3 and sys.version_info[1] >= 8:
                 asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
 
-        server_logger.info("Starting Gradio server")
-        interface = create_interface()
-        interface.launch(
-            server_name="0.0.0.0",
-            share=False,
-            debug=True,
-            prevent_thread_lock=True,  # Allow for proper cleanup
-            mcp_server=True,
-        )
+        server_logger.info("Starting server...")
+        
+        # Get port from environment variable or use default
+        port = int(os.getenv("PORT", 7860))
+        
+        # Run the FastAPI app with uvicorn
+        import uvicorn
+        uvicorn.run(app, host="0.0.0.0", port=port)
+        
     except Exception as e:
-        server_logger.error(f"Failed to start Gradio server: {str(e)}", exc_info=True)
+        server_logger.error(f"Failed to start server: {str(e)}", exc_info=True)
         raise