Initial commit V5

app.py

@@ -2,7 +2,20 @@
 import gradio as gr
 import uuid
 
+try:
+    from spaces import GPU
+    SPACES_GPU_AVAILABLE = True
+except ImportError:
+    # Not running on HF Spaces or spaces module not available
+    SPACES_GPU_AVAILABLE = False
+    GPU = None
+
 def create_mobile_optimized_interface():
+    """Create the mobile-optimized Gradio interface and return demo with components"""
+    
+    # Store components for wiring
+    interface_components = {}
+    
     with gr.Blocks(
         title="AI Research Assistant MVP",
         theme=gr.themes.Soft(
@@ -127,8 +140,10 @@ def create_mobile_optimized_interface():
                         show_label=False,
                         height="60vh",
                         elem_classes="chatbot-container",
+                        type="messages",
                         render=False  # Improve mobile performance
                     )
+                    interface_components['chatbot'] = chatbot
                     
                     # Mobile Input Area
                     with gr.Row():
@@ -140,6 +155,7 @@ def create_mobile_optimized_interface():
                             scale=4,
                             autofocus=True
                         )
+                        interface_components['message_input'] = message_input
                         
                         send_btn = gr.Button(
                             "↑ Send",
@@ -147,6 +163,7 @@ def create_mobile_optimized_interface():
                             scale=1,
                             min_width=80
                         )
+                        interface_components['send_btn'] = send_btn
                 
                 # Technical Details Tab (Collapsible for Mobile)
                 with gr.TabItem("🔍 Details", id="details_tab"):
@@ -209,8 +226,8 @@ def create_mobile_optimized_interface():
                 )
             
             gr.Button("Save Preferences", variant="primary")
-
-    return demo
+    
+    return demo, interface_components
 
 def setup_event_handlers(demo, event_handlers):
     """Setup event handlers for the interface"""
@@ -250,26 +267,44 @@ def simple_message_handler(message, chat_history):
     new_history = chat_history + [[message, response]]
     return new_history, ""
 
+def process_message(message, history):
+    """Process message with messages format"""
+    if not message.strip():
+        return history, ""
+    
+    response = f"I received your message: {message}. This is a placeholder response. The full agent system is ready to integrate!"
+    
+    # Add user and assistant messages
+    history.append((message, response))
+    
+    return history, ""
+
+# Decorate the chat handler with GPU if available
+if SPACES_GPU_AVAILABLE and GPU is not None:
+    @GPU  # This decorator is detected by HF Spaces for ZeroGPU allocation
+    def gpu_chat_handler(message, history):
+        """Handle chat messages with GPU support"""
+        return process_message(message, history)
+    chat_handler_fn = gpu_chat_handler
+else:
+    chat_handler_fn = process_message
+
 if __name__ == "__main__":
-    demo = create_mobile_optimized_interface()
+    demo, components = create_mobile_optimized_interface()
     
-    # Connect the UI components with the mock handler
-    # (In production, these would use the full orchestrator)
-    try:
-        # This assumes the demo is accessible - in Gradio 4.x, components are scoped
-        # For now, the UI will render even without handlers
-        demo.launch(
-            server_name="0.0.0.0",
-            server_port=7860,
-            share=False
-        )
-    except Exception as e:
-        print(f"Note: UI launched but handlers not connected yet: {e}")
-        print("The framework is ready for integration with the orchestrator.")
-        print("\nNext step: Connect to backend agents in main.py")
-        demo.launch(
-            server_name="0.0.0.0",
-            server_port=7860,
-            share=False
+    # Wire up the submit handler
+    if 'send_btn' in components and 'message_input' in components and 'chatbot' in components:
+        # Connect the submit handler with the GPU-decorated function
+        components['send_btn'].click(
+            fn=chat_handler_fn,
+            inputs=[components['message_input'], components['chatbot']],
+            outputs=[components['chatbot'], components['message_input']]
         )
+    
+    # Launch the app
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )