Update app.py

app.py

@@ -1,45 +1,96 @@
-import gradio as gr
-import requests
 import os
+import json
+import requests
+import gradio as gr
 
-API_BASE = os.getenv("ANALYSIS_API_URL", "https://baxin-flex-analysis.hf.space")
+API_BASE = os.getenv("ANALYSIS_API_URL", "https://baxin-opentrons-robot-analysis.hf.space").rstrip("/")
 TIMEOUT_SEC = float(os.getenv("HTTP_TIMEOUT", "30"))
 
-def analyze_protocol(text: str) -> str:
-    text = (text or "").strip()
-    if not text:
-        return "Empty protocol"
+def _pretty(obj) -> str:
+    try:
+        return json.dumps(obj, indent=2, ensure_ascii=False)
+    except Exception:
+        return str(obj)
+
+def _request(method: str, path: str, **kwargs) -> dict:
+    url = f"{API_BASE}{path}"
+    try:
+        resp = requests.request(method, url, timeout=TIMEOUT_SEC, **kwargs)
+    except requests.RequestException as e:
+        return {"ok": False, "url": url, "error": str(e)}
 
-    resp = requests.post(
-        f"{API_BASE}/analyze",
-        json={"protocol_text": text, "filename": "protocol.py"},
-        timeout=TIMEOUT_SEC,
-    )
+    out = {
+        "ok": resp.ok,
+        "url": url,
+        "status_code": resp.status_code,
+        "headers": dict(resp.headers),
+    }
     try:
-        payload = resp.json()
+        out["json"] = resp.json()
     except ValueError:
-        return f"Non-JSON response (HTTP {resp.status_code}):\n{resp.text[:2000]}"
+        out["text"] = resp.text[:4000]
+    return out
+
+def ping_root():
+    return _pretty(_request("GET", "/"))
+
+def ping_health():
+    return _pretty(_request("GET", "/health"))
+
+def ping_robot_health():
+    return _pretty(_request("GET", "/robot/health"))
 
-    if resp.status_code >= 400 or not payload.get("ok", True):
-        return f"Upload failed (HTTP {resp.status_code})\n{payload}"
+def analyze(protocol_text: str, filename: str):
+    protocol_text = (protocol_text or "").strip()
+    if not protocol_text:
+        return _pretty({"ok": False, "error": "Empty protocol_text"})
 
-    return str(payload)
+    payload = {"protocol_text": protocol_text, "filename": filename or "protocol.py"}
+    res = _request("POST", "/analyze", json=payload)
+    return _pretty(res)
 
-def send_message(text, history):
+def chat_send(text, history, filename):
     history = history or []
     history.append({"role": "user", "content": text})
-    result = analyze_protocol(text)
-    history.append({"role": "assistant", "content": result})
+
+    res = analyze(text, filename)
+    history.append({"role": "assistant", "content": res})
     return history
 
 with gr.Blocks() as app:
-    gr.Markdown("## Opentrons Protocol Analyzer (via FastAPI)")
-    textbox = gr.Textbox(lines=12, label="Paste protocol Python code")
-    send_button = gr.Button(value="Analyze")
-    chatbot = gr.Chatbot(label="Results")
-    clear_button = gr.ClearButton([textbox, chatbot])
-
-    send_button.click(send_message, [textbox, chatbot], [chatbot])
-    textbox.submit(send_message, [textbox, chatbot], [chatbot])
+    gr.Markdown("## Opentrons API Server Tester (FastAPI)")
+
+    with gr.Row():
+        api_url = gr.Textbox(value=API_BASE, label="API Base URL", interactive=False)
+        timeout = gr.Number(value=TIMEOUT_SEC, label="Timeout (sec)", interactive=False)
+
+    with gr.Row():
+        btn_root = gr.Button("GET /  (version/info)")
+        btn_health = gr.Button("GET /health")
+        btn_robot_health = gr.Button("GET /robot/health")
+
+    status_out = gr.Code(label="Response", language="json")
+
+    btn_root.click(fn=ping_root, inputs=[], outputs=[status_out])
+    btn_health.click(fn=ping_health, inputs=[], outputs=[status_out])
+    btn_robot_health.click(fn=ping_robot_health, inputs=[], outputs=[status_out])
+
+    gr.Markdown("### Analyze (/analyze)")
+    filename = gr.Textbox(value="protocol.py", label="filename")
+    protocol = gr.Textbox(lines=14, label="Paste protocol Python code")
+
+    with gr.Row():
+        btn_analyze = gr.Button("POST /analyze")
+        btn_clear = gr.Button("Clear")
+
+    btn_analyze.click(fn=analyze, inputs=[protocol, filename], outputs=[status_out])
+    btn_clear.click(fn=lambda: ("", ""), inputs=[], outputs=[protocol, status_out])
+
+    gr.Markdown("### Chat-style test")
+    chatbot = gr.Chatbot(label="History")
+    send_button = gr.Button(value="Analyze (chat)")
+
+    send_button.click(chat_send, [protocol, chatbot, filename], [chatbot])
+    protocol.submit(chat_send, [protocol, chatbot, filename], [chatbot])
 
 app.launch(server_name="0.0.0.0", server_port=int(os.getenv("PORT", "7860")))