Update app.py

app.py

@@ -1,18 +1,206 @@
+"""
+SuperCoder - Hugging Face Spaces Frontend (FIXED)
+Connects to your local API server via tunnel
+"""
 import gradio as gr
 import requests
+from typing import List, Tuple
+import os
 
-API_URL = "https://inge-chalcographic-helene.ngrok-free.dev"
+# ============================================================================
+# Configuration - YOUR TUNNEL URL
+# ============================================================================
+# The API_URL is pulled from a Space Secret for better security
+API_URL = os.getenv("API_URL", "https://your-tunnel-url.ngrok-free.dev")
+API_URL = API_URL.rstrip('/')
 
-def call_api(message):
-    response = requests.post(f"{API_URL}/api/chat", 
-                           json={"messages": [{"role": "user", "content": message}]})
-    return response.json().get("response", "Error")
+# ============================================================================
+# API Client Functions
+# ============================================================================
+def call_api(message: str, temperature: float = 0.1, max_tokens: int = 512) -> str:
+    """Call the remote API with error handling"""
+    if not message or not message.strip():
+        return "⚠️ Please enter a message"
+    
+    try:
+        response = requests.post(
+            f"{API_URL}/api/chat",
+            json={
+                "messages": [{"role": "user", "content": message}],
+                "temperature": temperature,
+                "max_tokens": max_tokens
+            },
+            timeout=90,
+            headers={
+                "Content-Type": "application/json",
+                "ngrok-skip-browser-warning": "true"
+            }
+        )
+        
+        if response.status_code == 200:
+            result = response.json()
+            return result.get("response", "No response from API")
+        elif response.status_code == 503:
+            return "🔧 Backend service unavailable. Please ensure your local server is running."
+        else:
+            return f"❌ API Error ({response.status_code}): {response.text[:200]}"
+            
+    except requests.exceptions.Timeout:
+        return "⏱️ Request timed out. The model might be processing a complex request or the server is down."
+    except requests.exceptions.ConnectionError:
+        return f"🔌 Cannot connect to API at {API_URL}. Please verify:\n1. Local server is running\n2. Tunnel (ngrok/cloudflare) is active\n3. API_URL is correct"
+    except Exception as e:
+        return f"⚠️ Unexpected error: {str(e)}"
 
-with gr.Blocks() as demo:
-    gr.Markdown("# AI Assistant")
-    chatbot = gr.Chatbot()
-    msg_input = gr.Textbox(placeholder="Ask me about code...")
-    msg_input.submit(lambda m, h: (h + [(m, call_api(m))], ""), 
-                     [msg_input, chatbot], [chatbot, msg_input])
+def check_api_status() -> str:
+    """Check if the API is reachable and healthy"""
+    try:
+        response = requests.get(
+            f"{API_URL}/health", 
+            timeout=10, # Increased timeout slightly for cold starts
+            headers={"ngrok-skip-browser-warning": "true"}
+        )
+        
+        if response.status_code == 200:
+            data = response.json()
+            status = data.get("status", "unknown")
+            if status == "ok":
+                return "✅ Connected - Backend Ready"
+            else:
+                return f"⚠️ Connected but status: {status}"
+        else:
+            return f"❌ API returned status {response.status_code}"
+            
+    except requests.exceptions.ConnectionError:
+        return f"🔴 Cannot reach {API_URL} - Check tunnel status"
+    except requests.exceptions.Timeout:
+        return "⏱️ Health check timed out"
+    except Exception as e:
+        return f"❌ Error: {str(e)}"
 
-demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
+# ============================================================================
+# Gradio Interface
+# ============================================================================
+def respond(message: str, history: List[Tuple[str, str]], temperature: float, max_tokens: int):
+    """Handle chat responses"""
+    history.append((message, None))
+    yield history # Show user message immediately
+
+    bot_response = call_api(message, temperature, max_tokens)
+    history[-1] = (message, bot_response)
+    yield history
+
+def apply_template(template: str, history: List[Tuple[str, str]]) -> tuple:
+    """Apply a code template"""
+    templates = {
+        "Explain Code": "Please explain the following code in detail:\n```python\n# Paste your code here\n```",
+        "Debug Code": "I have a bug in my code. Can you help me debug it?\n```python\n# Paste your buggy code here\n```",
+        "Write Function": "Please write a Python function that: [describe what you need]",
+        "Optimize Code": "Can you optimize this code for better performance?\n```python\n# Paste your code here\n```",
+        "Add Comments": "Please add clear comments to this code:\n```python\n# Paste your code here\n```"
+    }
+    return templates.get(template, ""), history
+
+# Create the Gradio interface
+with gr.Blocks(
+    title="SuperCoder Pro", 
+    theme=gr.themes.Soft(primary_hue="indigo"),
+    css=".gradio-container {max-width: 1200px !important}"
+) as demo:
+    
+    gr.Markdown(
+        """
+        # 🤖 SuperCoder Pro
+        ### AI-Powered Coding Assistant
+        > **Note:** This interface connects to a local backend via a secure tunnel. 
+        > Ensure your local server and tunnel are running.
+        """
+    )
+    
+    # Status bar
+    with gr.Row():
+        with gr.Column(scale=4):
+            # --- FIX ---: Set a static default value here
+            status_display = gr.Textbox(
+                value="⏳ Initializing...", 
+                label="🔌 Backend Status",
+                interactive=False,
+                show_copy_button=True
+            )
+        with gr.Column(scale=1):
+            refresh_btn = gr.Button("🔄 Refresh Status", size="sm")
+    
+    # Main chat interface
+    with gr.Row():
+        with gr.Column(scale=3):
+            chatbot = gr.Chatbot(
+                label="💬 Conversation",
+                height=500,
+                show_copy_button=True,
+                avatar_images=(None, "🤖"),
+                bubble_full_width=False
+            )
+            
+            with gr.Row():
+                msg_input = gr.Textbox(
+                    placeholder="Ask me to write, explain, debug, or review code...",
+                    scale=5,
+                    lines=2,
+                    show_label=False,
+                    autofocus=True,
+                    container=False
+                )
+                send_btn = gr.Button("Send 🚀", scale=1, variant="primary")
+        
+        # Settings sidebar
+        with gr.Column(scale=1):
+            gr.Markdown("### ⚙️ Model Settings")
+            temperature = gr.Slider(0.0, 1.0, value=0.1, step=0.05, label="🌡️ Temperature")
+            max_tokens = gr.Slider(128, 4096, value=1024, step=128, label="📏 Max Tokens")
+            
+            gr.Markdown("---")
+            gr.Markdown("### 🎯 Quick Templates")
+            template_dropdown = gr.Dropdown(
+                choices=["Explain Code", "Debug Code", "Write Function", "Optimize Code", "Add Comments"],
+                label="Select Template",
+                value="Explain Code"
+            )
+            use_template_btn = gr.Button("📝 Use Template", size="sm")
+            clear_btn = gr.Button("🗑️ Clear Chat", variant="stop", size="sm")
+            
+            gr.Markdown("---")
+            gr.Markdown(f"""### 📡 Connection Info\n**API Endpoint:**\n`{API_URL}`""")
+    
+    # Event handlers
+    
+    # --- FIX ---: Use the demo.load() event to check status after UI is ready
+    demo.load(check_api_status, outputs=[status_display])
+
+    refresh_btn.click(check_api_status, outputs=[status_display])
+
+    msg_submit_event = msg_input.submit(
+        respond,
+        inputs=[msg_input, chatbot, temperature, max_tokens],
+        outputs=[chatbot]
+    )
+    msg_submit_event.then(lambda: gr.update(value=""), outputs=[msg_input])
+
+    send_btn.click(
+        respond,
+        inputs=[msg_input, chatbot, temperature, max_tokens],
+        outputs=[chatbot]
+    ).then(lambda: gr.update(value=""), outputs=[msg_input])
+    
+    use_template_btn.click(apply_template, inputs=[template_dropdown, chatbot], outputs=[msg_input, chatbot])
+    
+    clear_btn.click(lambda: [], outputs=[chatbot])
+
+# ============================================================================
+# Launch Configuration for HF Spaces
+# ============================================================================
+if __name__ == "__main__":
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        show_error=True
+    )