ts startup

app/main.py

@@ -387,7 +387,7 @@ def process_query_core(
         else:
             return f"Error: {str(e)}"
 
-# PROBABLY REMOVE - GRADIO VERSION (unless you need the web interface for debugging)
+
 async def process_query_gradio_streaming(query: str, file_upload, reports_filter: str = "", sources_filter: str = "", 
                                  subtype_filter: str = "", year_filter: str = ""):
     """Streaming version of process_query_gradio that yields partial results"""
@@ -469,7 +469,7 @@ async def process_query_gradio_streaming(query: str, file_upload, reports_filter
         logger.error(f"Streaming pipeline failed: {str(e)}")
         yield {"type": "error", "content": f"Error: {str(e)}"}
 
-# TODO: PROBABLY REMOVE - SEEMS REDUNDANT
+
 async def chatui_adapter(data):
     """Updated to return content without SSE formatting for LangServe"""
     try:
@@ -547,36 +547,36 @@ def process_query_langserve(input_data: ChatFedInput) -> ChatFedOutput:
     return ChatFedOutput(result=result["result"], metadata=result["metadata"])
 
 # GRADIO TEST UI
-def create_gradio_interface():
-    with gr.Blocks(title="ChatFed Orchestrator") as demo:
-        gr.Markdown("# ChatFed Orchestrator")
-        gr.Markdown("Upload documents (PDF/DOCX/GeoJSON) alongside your queries for enhanced context. MCP endpoints available at `/gradio_api/mcp/sse`")
-        
-        with gr.Row():
-            with gr.Column():
-                query_input = gr.Textbox(label="Query", lines=2, placeholder="Enter your question...")
-                file_input = gr.File(label="Upload Document (PDF/DOCX/GeoJSON)", file_types=[".pdf", ".docx", ".geojson", ".json"])
+# def create_gradio_interface():
+#     with gr.Blocks(title="ChatFed Orchestrator") as demo:
+#         gr.Markdown("# ChatFed Orchestrator")
+#         gr.Markdown("Upload documents (PDF/DOCX/GeoJSON) alongside your queries for enhanced context. MCP endpoints available at `/gradio_api/mcp/sse`")
+        
+#         with gr.Row():
+#             with gr.Column():
+#                 query_input = gr.Textbox(label="Query", lines=2, placeholder="Enter your question...")
+#                 file_input = gr.File(label="Upload Document (PDF/DOCX/GeoJSON)", file_types=[".pdf", ".docx", ".geojson", ".json"])
                 
-                with gr.Accordion("Filters (Optional)", open=False):
-                    reports_filter_input = gr.Textbox(label="Reports Filter", placeholder="e.g., annual_reports")
-                    sources_filter_input = gr.Textbox(label="Sources Filter", placeholder="e.g., internal")
-                    subtype_filter_input = gr.Textbox(label="Subtype Filter", placeholder="e.g., financial")
-                    year_filter_input = gr.Textbox(label="Year Filter", placeholder="e.g., 2024")
+#                 with gr.Accordion("Filters (Optional)", open=False):
+#                     reports_filter_input = gr.Textbox(label="Reports Filter", placeholder="e.g., annual_reports")
+#                     sources_filter_input = gr.Textbox(label="Sources Filter", placeholder="e.g., internal")
+#                     subtype_filter_input = gr.Textbox(label="Subtype Filter", placeholder="e.g., financial")
+#                     year_filter_input = gr.Textbox(label="Year Filter", placeholder="e.g., 2024")
                 
-                submit_btn = gr.Button("Submit", variant="primary")
+#                 submit_btn = gr.Button("Submit", variant="primary")
             
-            with gr.Column():
-                output = gr.Textbox(label="Response", lines=15, show_copy_button=True)
-        
-        # Use streaming function
-        submit_btn.click(
-            fn=process_query_gradio_streaming,
-            inputs=[query_input, file_input, reports_filter_input, sources_filter_input, 
-                   subtype_filter_input, year_filter_input],
-            outputs=output
-        )
+#             with gr.Column():
+#                 output = gr.Textbox(label="Response", lines=15, show_copy_button=True)
+        
+#         # Use streaming function
+#         submit_btn.click(
+#             fn=process_query_gradio_streaming,
+#             inputs=[query_input, file_input, reports_filter_input, sources_filter_input, 
+#                    subtype_filter_input, year_filter_input],
+#             outputs=output
+#         )
     
-    return demo
+#     return demo
 
 @asynccontextmanager
 async def lifespan(app: FastAPI):
@@ -798,16 +798,16 @@ add_routes(
 )
 
 if __name__ == "__main__":
-    # Create Gradio interface
-    demo = create_gradio_interface()
+    # # Create Gradio interface
+    # demo = create_gradio_interface()
     
     # Mount Gradio app to FastAPI
-    app = gr.mount_gradio_app(app, demo, path="/gradio")
+    # app = gr.mount_gradio_app(app, demo, path="/gradio")
     
-    host = os.getenv("HOST", "localhost")
+    host = os.getenv("HOST", "0.0.0.0")  
     port = int(os.getenv("PORT", "7860"))
     
     logger.info(f"Starting FastAPI server on {host}:{port}")
-    logger.info(f"Gradio UI available at: http://{host}:{port}/gradio")  # Remove if Gradio is removed
+    # logger.info(f"Gradio UI available at: http://{host}:{port}/gradio") 
     
     uvicorn.run(app, host=host, port=port, log_level="info", access_log=True)