Update app.py

app.py

@@ -1,56 +1,67 @@
 import gradio as gr
+import subprocess
+import threading
+import time
 import requests
 import json
 from datetime import datetime
 import os
-import subprocess
-import threading
-import time
 
-# Configuration - for HuggingFace Spaces
-MCP_SERVER_URL = "https://elanuk-mcp-hf.hf.space/"  # Local server URL
+# Configuration
+MCP_SERVER_URL = "https://elanuk-mcp-hf.hf.space/"
+
+# Bihar districts list
+BIHAR_DISTRICTS = [
+    "Patna", "Gaya", "Bhagalpur", "Muzaffarpur", "Darbhanga", "Siwan", 
+    "Begusarai", "Katihar", "Nalanda", "Rohtas", "Saran", "Samastipur",
+    "Madhubani", "Purnia", "Araria", "Kishanganj", "Supaul", "Madhepura",
+    "Saharsa", "Khagaria", "Munger", "Lakhisarai", "Sheikhpura", "Nawada",
+    "Jamui", "Jehanabad", "Aurangabad", "Arwal", "Kaimur", "Buxar",
+    "Bhojpur", "Saran", "Siwan", "Gopalganj", "East Champaran", "West Champaran",
+    "Sitamarhi", "Sheohar", "Vaishali"
+]
 
 def start_mcp_server():
     """Start the MCP server in background"""
     try:
-        # Start mcp_server.py as subprocess  
+        print("🚀 Starting MCP Server...")
+        # Set environment variable for the server port
+        env = os.environ.copy()
+        env["PORT"] = str(MCP_SERVER_PORT)
+        
         process = subprocess.Popen(
             ["python", "mcp_server.py"],
             stdout=subprocess.PIPE,
-            stderr=subprocess.PIPE
+            stderr=subprocess.PIPE,
+            env=env,
+            text=True
         )
         
-        # Wait a bit for server to start
-        time.sleep(10)
+        # Wait for server to start
+        print("⏳ Waiting for server to start...")
+        time.sleep(15)
+        
+        # Check if server is running
+        try:
+            response = requests.get(f"{MCP_SERVER_URL}/api/health", timeout=5)
+            if response.status_code == 200:
+                print("✅ MCP Server started successfully!")
+                return process
+        except:
+            pass
+            
+        print("⚠️ Server may still be starting...")
         return process
+        
     except Exception as e:
-        print(f"Failed to start MCP server: {e}")
+        print(f"❌ Failed to start MCP server: {e}")
         return None
 
-# Start server in background thread
-def start_server_thread():
-    start_mcp_server()
-
-server_thread = threading.Thread(target=start_server_thread, daemon=True)
-server_thread.start()
-
-# Bihar districts list
-BIHAR_DISTRICTS = [
-    "Patna", "Gaya", "Bhagalpur", "Muzaffarpur", "Darbhanga", "Siwan", 
-    "Begusarai", "Katihar", "Nalanda", "Rohtas", "Saran", "Samastipur",
-    "Madhubani", "Purnia", "Araria", "Kishanganj", "Supaul", "Madhepura",
-    "Saharsa", "Khagaria", "Munger", "Lakhisarai", "Sheikhpura", "Nawada",
-    "Jamui", "Jehanabad", "Aurangabad", "Arwal", "Kaimur", "Buxar",
-    "Bhojpur", "Saran", "Siwan", "Gopalganj", "East Champaran", "West Champaran",
-    "Sitamarhi", "Sheohar", "Vaishali"
-]
-
 def format_workflow_output(raw_output):
     """Format the workflow output for better display"""
     if not raw_output:
         return "❌ No output received"
     
-    # Split into lines and format
     lines = raw_output.split('\n')
     formatted_lines = []
     
@@ -60,11 +71,10 @@ def format_workflow_output(raw_output):
             formatted_lines.append("")
             continue
             
-        # Format headers
         if line.startswith('🌾') and 'Workflow' in line:
             formatted_lines.append(f"## {line}")
         elif line.startswith('=') or line.startswith('-'):
-            continue  # Skip separator lines
+            continue
         elif line.startswith('🌤️') or line.startswith('✅ Workflow'):
             formatted_lines.append(f"### {line}")
         elif line.startswith('📱') or line.startswith('📞') or line.startswith('🎙️') or line.startswith('🤖'):
@@ -120,7 +130,6 @@ def test_mcp_workflow(district):
         return "❌ Please select a district", "", ""
     
     try:
-        # Make request to MCP server
         payload = {
             "state": "bihar",
             "district": district.lower()
@@ -129,17 +138,14 @@ def test_mcp_workflow(district):
         response = requests.post(
             f"{MCP_SERVER_URL}/api/run-workflow",
             json=payload,
-            timeout=30
+            timeout=60  # Increased timeout
         )
         
         if response.status_code == 200:
             result = response.json()
             
-            # Format outputs
             workflow_output = format_workflow_output(result.get('message', ''))
             alert_summary = format_alert_summary(result.get('raw_data', {}))
-            
-            # Create CSV download content
             csv_content = result.get('csv', '')
             
             return workflow_output, alert_summary, csv_content
@@ -151,7 +157,7 @@ def test_mcp_workflow(district):
     except requests.exceptions.Timeout:
         return "⏰ Request timed out. The server might be processing...", "", ""
     except requests.exceptions.ConnectionError:
-        return f"🔌 Connection Error: Cannot reach MCP server at {MCP_SERVER_URL}", "", ""
+        return f"🔌 Connection Error: Cannot reach MCP server. Is it running?", "", ""
     except Exception as e:
         return f"❌ Error: {str(e)}", "", ""
 
@@ -164,8 +170,19 @@ def check_server_health():
             return f"✅ Server Online | OpenAI: {'✅' if data.get('openai_available') else '❌'} | Time: {data.get('timestamp', 'N/A')}"
         else:
             return f"⚠️ Server responded with status {response.status_code}"
-    except:
-        return f"❌ Server Offline or Unreachable ({MCP_SERVER_URL})"
+    except Exception as e:
+        return f"❌ Server Offline: {str(e)}"
+
+# Start server in background thread
+print("🔧 Initializing BIHAR AgMCP...")
+server_process = None
+
+def start_server_thread():
+    global server_process
+    server_process = start_mcp_server()
+
+server_thread = threading.Thread(target=start_server_thread, daemon=True)
+server_thread.start()
 
 # Create Gradio interface
 with gr.Blocks(
@@ -176,11 +193,6 @@ with gr.Blocks(
         max-width: 1200px;
         margin: auto;
     }
-    .status-box {
-        padding: 10px;
-        border-radius: 5px;
-        margin: 10px 0;
-    }
     """
 ) as demo:
     
@@ -189,32 +201,32 @@ with gr.Blocks(
     
     **AI-Powered Weather Alerts for Bihar Farmers**
     
-    This interface tests the MCP (Model Context Protocol) server that generates personalized weather alerts for agricultural activities in Bihar districts.
+    This system generates personalized weather alerts for agricultural activities in Bihar districts.
     
     ## 📋 How to Use:
-    1. **Select District**: Choose a Bihar district from the dropdown
-    2. **Run Workflow**: Click the button to generate weather alerts
-    3. **View Results**: See formatted workflow output and alert summary
-    4. **Download Data**: Get CSV export of the alert data
+    1. **Wait for Server**: Ensure server status shows "Online" below
+    2. **Select District**: Choose a Bihar district from the dropdown  
+    3. **Run Workflow**: Click the button to generate weather alerts
+    4. **View Results**: See formatted workflow output and alert summary
+    5. **Download Data**: Get CSV export of the alert data
     
-    The system will:
+    The system will automatically:
     - Select a random village in the district
-    - Choose appropriate crops based on season and region
+    - Choose appropriate crops based on season and region  
     - Generate weather-based agricultural alerts
-    - Create messages for multiple communication channels (SMS, WhatsApp, etc.)
+    - Create messages for multiple communication channels
     """)
     
     # Server status
     with gr.Row():
         server_status = gr.Textbox(
             label="🔧 Server Status", 
-            value=check_server_health(),
+            value="🔄 Starting server...",
             interactive=False,
             container=True
         )
         
-        refresh_btn = gr.Button("🔄 Refresh Status", size="sm")
-        refresh_btn.click(check_server_health, outputs=server_status)
+        refresh_btn = gr.Button("🔄 Check Status", size="sm")
     
     # Main interface
     with gr.Row():
@@ -234,11 +246,11 @@ with gr.Blocks(
             
             gr.Markdown("""
             ### 💡 What happens next?
-            - Weather data collection
-            - Crop stage estimation  
+            - Weather data collection from multiple sources
+            - Intelligent crop stage estimation  
             - AI-powered alert generation
-            - Multi-channel message creation
-            - CSV data export
+            - Multi-channel message creation (SMS, WhatsApp, etc.)
+            - Comprehensive CSV data export
             """)
     
     # Results section
@@ -246,13 +258,13 @@ with gr.Blocks(
         with gr.Column(scale=2):
             workflow_output = gr.Markdown(
                 label="📋 Workflow Output",
-                value="Click 'Generate Weather Alert' to see results..."
+                value="Server is starting... Please wait and check server status above."
             )
         
         with gr.Column(scale=1):
             alert_summary = gr.Markdown(
-                label="📊 Alert Summary",
-                value="Alert details will appear here..."
+                label="📊 Alert Summary", 
+                value="Alert details will appear here after running workflow..."
             )
     
     # CSV export
@@ -266,46 +278,46 @@ with gr.Blocks(
     def run_workflow_with_csv(district):
         workflow, summary, csv_content = test_mcp_workflow(district)
         
-        # Create temporary CSV file if content exists
-        if csv_content:
+        if csv_content and not csv_content.startswith("Error"):
             filename = f"bihar_alert_{district.lower()}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv"
-            with open(filename, 'w') as f:
+            with open(filename, 'w', encoding='utf-8') as f:
                 f.write(csv_content)
             return workflow, summary, gr.File(value=filename, visible=True)
         else:
             return workflow, summary, gr.File(visible=False)
     
+    # Connect events
+    refresh_btn.click(check_server_health, outputs=server_status)
     run_btn.click(
         run_workflow_with_csv,
-        inputs=[district_input],
+        inputs=[district_input], 
         outputs=[workflow_output, alert_summary, csv_output]
     )
     
+    # Auto-refresh server status after a delay
+    def auto_check_status():
+        time.sleep(20)  # Wait 20 seconds
+        return check_server_health()
+    
+    demo.load(auto_check_status, outputs=server_status)
+    
     # Footer
     gr.Markdown("""
     ---
     
     ### 🔗 System Information:
-    - **State Coverage**: Bihar (38 districts)
+    - **State Coverage**: Bihar (38+ districts)
     - **Crops Supported**: Rice, Wheat, Maize, Sugarcane, Mustard, and more
     - **Weather Sources**: Open-Meteo API with AI enhancement
     - **Communication Channels**: SMS, WhatsApp, USSD, IVR, Telegram
     
-    ### 📞 Agent Outputs:
-    The system generates formatted messages for:
-    - **📱 SMS**: Short text alerts for basic phones
-    - **📱 WhatsApp**: Rich media messages with emojis
-    - **📞 USSD**: Interactive menu systems
-    - **🎙️ IVR**: Voice script for phone calls  
-    - **🤖 Telegram**: Bot-friendly formatted messages
-    
     *Built with MCP (Model Context Protocol) for agricultural intelligence*
     """)
 
-# Launch configuration for HuggingFace Spaces
+# Launch the interface
 if __name__ == "__main__":
+    print("🌾 Launching BIHAR AgMCP Interface...")
     demo.launch(
-        share=True,
         server_name="0.0.0.0",
         server_port=7860,
         show_error=True