Upload app.py with huggingface_hub

app.py

@@ -1,59 +1,41 @@
-import gradio as gr
-import requests
-import json
-
-API_URL = "http://94.205.175.139:8081"
-API_KEY="d"a-demo-key-0001"
-
-def get_status():
-    try:
-        r = requests.get(f"{API_URL}/", timeout=5)
-        data = r.json()
-        return f"Status: {data.get('status', 'unknown')}
-Phones: {data.get('phones_connected', 0)}
-Models: {len(data.get('models', []))}
-Pricing: {data.get('pricing', {})}"
-    except Exception as e:
-        return f"API Error: {e}"
-
-def chat(message, model):
-    try:
-        r = requests.post(
-            f"{API_URL}/v1/chat/completions",
-            headers={"Authorization": f"Bearer {API_KEY}", "Content-Type": "application/json"},
-            json={"model": model, "messages": [{"role": "user", "content": message}], "max_tokens": 50},
-            timeout=60
-        )
-        data = r.json()
-        content = data.get("choices", [{}])[0].get("message", {}).get("content", "No response")
-        phone = data.get("phone_info", {})
-        tps = phone.get("generation_tps", 0)
-        serial = phone.get("serial", "unknown")
-        return f"{content}
-
----
-Phone: {serial} | Speed: {tps} t/s"
-    except Exception as e:
-        return f"Error: {e}"
-
-with gr.Blocks(title="dispatchAI Phone Farm Monitor") as app:
-    gr.Markdown("# dispatchAI Phone Farm Monitor")
-    with gr.Row():
-        status_btn = gr.Button("Refresh Status")
-        status_out = gr.Textbox(label="Status", lines=5)
-    status_btn.click(get_status, outputs=status_out)
-    
-    gr.Markdown("## Test Chat")
-    with gr.Row():
-        model = gr.Dropdown([
-            "dispatchAI/SmolLM2-135M-Instruct-mobile",
-            "dispatchAI/Qwen2.5-0.5B-Instruct-mobile-int4",
-            "dispatchAI/Llama-3.2-1B-Instruct-Q4-mobile"
-        ], label="Model", value="dispatchAI/SmolLM2-135M-Instruct-mobile")
-    with gr.Row():
-        msg = gr.Textbox(label="Message", value="What is the capital of France?")
-        send = gr.Button("Send")
-    out = gr.Textbox(label="Response", lines=5)
-    send.click(chat, inputs=[msg, model], outputs=out)
-
-app.launch()
+import gradio as gr
+import httpx
+import asyncio
+import json
+
+BASE = "http://127.0.0.1:8081"
+
+def get_status():
+    try:
+        r = httpx.get(f"{BASE}/", timeout=5.0)
+        data = r.json()
+        return f"API: {data["status"]}
+Phones: {data["phones_connected"]}
+Models: {len(data["models"])}"
+    except:
+        return "API offline"
+
+def chat(message):
+    try:
+        r = httpx.post(f"{BASE}/v1/chat/completions",
+            headers={"Authorization": "Bearer da-demo-key-0001", "Content-Type": "application/json"},
+            json={"model": "dispatchAI/SmolLM2-135M-Instruct-mobile", "messages": [{"role": "user", "content": message}], "max_tokens": 50},
+            timeout=120.0)
+        data = r.json()
+        return data["choices"][0]["message"]["content"]
+    except Exception as e:
+        return f"Error: {e}"
+
+with gr.Blocks(title="dispatchAI Monitor") as app:
+    gr.Markdown("# dispatchAI Phone Farm Monitor")
+    with gr.Row():
+        status_btn = gr.Button("Check Status")
+        status_out = gr.Textbox(label="Status")
+    status_btn.click(get_status, outputs=status_out)
+    with gr.Row():
+        msg = gr.Textbox(label="Message", placeholder="Ask a question...")
+        send_btn = gr.Button("Send")
+        response = gr.Textbox(label="Response")
+    send_btn.click(chat, inputs=msg, outputs=response)
+
+app.launch()