Upload gradio_app.py

gradio_app.py

@@ -0,0 +1,277 @@
+"""
+🤖 PENNY Gradio Interface
+Hugging Face Space Entry Point
+
+This file connects PENNY's FastAPI backend to a Gradio chat interface,
+allowing users to interact with Penny through a web UI on Hugging Face Spaces.
+"""
+
+import gradio as gr
+import logging
+import sys
+from typing import List, Tuple
+from datetime import datetime
+
+# Setup logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+    handlers=[logging.StreamHandler(sys.stdout)]
+)
+logger = logging.getLogger(__name__)
+
+# Import Penny's orchestrator
+try:
+    from app.orchestrator import run_orchestrator, get_service_availability
+    from app.location_utils import SupportedCities
+    logger.info("✅ Successfully imported PENNY modules")
+except ImportError as e:
+    logger.error(f"❌ Failed to import PENNY modules: {e}")
+    raise
+
+# ============================================================
+# GRADIO CHAT INTERFACE
+# ============================================================
+
+async def chat_with_penny(
+    message: str,
+    city: str,
+    history: List[Tuple[str, str]]
+) -> Tuple[List[Tuple[str, str]], str]:
+    """
+    Process user message through PENNY's orchestrator and return response.
+    
+    Args:
+        message: User's input text
+        city: Selected city/tenant_id
+        history: Chat history (list of (user_msg, bot_msg) tuples)
+    
+    Returns:
+        Tuple of (updated_history, empty_string_to_clear_input)
+    """
+    if not message.strip():
+        return history, ""
+    
+    try:
+        # Build context from selected city
+        context = {}
+        if city and city != "Not sure / Other":
+            context["tenant_id"] = city
+        
+        logger.info(f"Processing message: '{message[:50]}...' for city: {city}")
+        
+        # Call PENNY's orchestrator
+        result = await run_orchestrator(message, context)
+        
+        # Extract response
+        reply = result.get("reply", "I'm having trouble right now. Please try again! 💛")
+        
+        # Add to history
+        history.append((message, reply))
+        
+        logger.info(f"Response generated successfully (intent: {result.get('intent')})")
+        
+        return history, ""
+        
+    except Exception as e:
+        logger.error(f"Error processing message: {e}", exc_info=True)
+        error_reply = (
+            "I'm having trouble processing your request right now. "
+            "Please try again in a moment! 💛"
+        )
+        history.append((message, error_reply))
+        return history, ""
+
+
+def get_city_choices() -> List[str]:
+    """Get list of supported cities for dropdown."""
+    try:
+        cities = SupportedCities.get_all_cities()
+        city_names = [city.full_name for city in cities]
+        return ["Not sure / Other"] + sorted(city_names)
+    except Exception as e:
+        logger.error(f"Error loading cities: {e}")
+        return ["Not sure / Other", "Atlanta, GA", "Seattle, WA"]
+
+
+def get_service_status() -> str:
+    """Display current service availability status."""
+    try:
+        services = get_service_availability()
+        status_lines = ["**PENNY Service Status:**\n"]
+        
+        for service, available in services.items():
+            icon = "✅" if available else "⚠️"
+            status = "Available" if available else "Limited"
+            status_lines.append(f"{icon} {service.replace('_', ' ').title()}: {status}")
+        
+        return "\n".join(status_lines)
+    except Exception as e:
+        logger.error(f"Error getting service status: {e}")
+        return "**Status:** Unable to check service availability"
+
+
+# ============================================================
+# GRADIO UI DEFINITION
+# ============================================================
+
+# Custom CSS for a polished look
+custom_css = """
+#chatbot {
+    height: 500px;
+    overflow-y: auto;
+}
+.gradio-container {
+    font-family: 'Inter', sans-serif;
+}
+footer {
+    display: none !important;
+}
+"""
+
+# Build the Gradio interface
+with gr.Blocks(
+    theme=gr.themes.Soft(primary_hue="amber", secondary_hue="blue"),
+    css=custom_css,
+    title="PENNY - Civic Assistant"
+) as demo:
+    
+    # Header
+    gr.Markdown(
+        """
+        # 🤖 PENNY - People's Engagement Network Navigator for You
+        
+        **Your multilingual civic assistant connecting residents to local services.**
+        
+        Ask me about:
+        - 🌤️ Weather conditions
+        - 📅 Community events
+        - 🏛️ Local resources (shelters, libraries, food banks)
+        - 🌍 Translation (27 languages)
+        - 📄 Document help
+        """
+    )
+    
+    with gr.Row():
+        with gr.Column(scale=2):
+            # City selector
+            city_dropdown = gr.Dropdown(
+                choices=get_city_choices(),
+                value="Not sure / Other",
+                label="📍 Select Your City",
+                info="Choose your city for location-specific information"
+            )
+            
+            # Chat interface
+            chatbot = gr.Chatbot(
+                label="Chat with PENNY",
+                elem_id="chatbot",
+                avatar_images=(None, "🤖"),
+                show_label=False
+            )
+            
+            # Input row
+            with gr.Row():
+                msg_input = gr.Textbox(
+                    placeholder="Type your message here... (e.g., 'What's the weather today?')",
+                    show_label=False,
+                    scale=4,
+                    container=False
+                )
+                submit_btn = gr.Button("Send", variant="primary", scale=1)
+            
+            # Example queries
+            gr.Examples(
+                examples=[
+                    ["What's the weather today?"],
+                    ["Any events this weekend?"],
+                    ["I need help finding a library"],
+                    ["Show me local resources"],
+                    ["Translate 'hello' to Spanish"]
+                ],
+                inputs=msg_input,
+                label="💡 Try asking:"
+            )
+        
+        with gr.Column(scale=1):
+            # Service status panel
+            status_display = gr.Markdown(
+                value=get_service_status(),
+                label="System Status"
+            )
+            
+            gr.Markdown(
+                """
+                ### 🌟 Features
+                
+                - **27 Languages** supported
+                - **Real-time weather** via Azure Maps
+                - **Community events** database
+                - **Local resource** finder
+                - **Document processing** help
+                
+                ---
+                
+                ### 📍 Supported Cities
+                
+                - Atlanta, GA
+                - Birmingham, AL
+                - Chesterfield, VA
+                - El Paso, TX
+                - Providence, RI
+                - Seattle, WA
+                
+                ---
+                
+                💛 *PENNY is here to help connect you with civic resources!*
+                """
+            )
+    
+    # Event handlers
+    submit_btn.click(
+        fn=chat_with_penny,
+        inputs=[msg_input, city_dropdown, chatbot],
+        outputs=[chatbot, msg_input]
+    )
+    
+    msg_input.submit(
+        fn=chat_with_penny,
+        inputs=[msg_input, city_dropdown, chatbot],
+        outputs=[chatbot, msg_input]
+    )
+    
+    # Footer
+    gr.Markdown(
+        """
+        ---
+        **Built with:** FastAPI • Gradio • Azure ML • Transformers
+        """
+    )
+
+
+# ============================================================
+# LAUNCH
+# ============================================================
+
+if __name__ == "__main__":
+    logger.info("=" * 60)
+    logger.info("🚀 Launching PENNY Gradio Interface")
+    logger.info("=" * 60)
+    
+    # Display service availability at startup
+    logger.info("\n📊 Service Availability Check:")
+    services = get_service_availability()
+    for service, available in services.items():
+        status = "✅" if available else "❌"
+        logger.info(f"  {status} {service}: {'Available' if available else 'Not loaded'}")
+    
+    logger.info("\n" + "=" * 60)
+    logger.info("🤖 PENNY is ready to help residents!")
+    logger.info("=" * 60 + "\n")
+    
+    # Launch the Gradio app
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )