Fix API: use Textbox.submit with api_name for reliable endpoint

app.py

@@ -10,14 +10,13 @@ DATASET_REPO = "bobber/routangseng-telemetry"
 HF_TOKEN = os.environ.get("HF_TOKEN")
 BUFFER = []
 BUFFER_LOCK = Lock()
-FLUSH_INTERVAL = 60  # seconds
-FLUSH_SIZE = 10  # or flush when buffer hits this size
+FLUSH_INTERVAL = 60
+FLUSH_SIZE = 10
 
 api = HfApi(token=HF_TOKEN)
 
 
 def flush_buffer():
-    """Append buffered entries to the dataset."""
     global BUFFER
     with BUFFER_LOCK:
         if not BUFFER:
@@ -46,26 +45,23 @@ def flush_buffer():
 
 
 def periodic_flush():
-    """Flush buffer periodically."""
     try:
         n = flush_buffer()
         if n > 0:
-            print(f"[telemetry] Flushed {n} entries to dataset")
+            print(f"[telemetry] Flushed {n} entries")
     except Exception as e:
         print(f"[telemetry] Flush error: {e}")
     Timer(FLUSH_INTERVAL, periodic_flush).start()
 
 
-# Start periodic flush
 Timer(FLUSH_INTERVAL, periodic_flush).start()
 
 
-def collect(data: str) -> dict:
-    """Receive telemetry JSON from the WebGPU Space."""
+def collect(data: str) -> str:
     try:
         entry = json.loads(data)
         if "question" not in entry or "answer" not in entry:
-            return {"status": "error", "message": "missing question or answer"}
+            return json.dumps({"status": "error", "message": "missing fields"})
 
         entry["received_at"] = datetime.utcnow().isoformat() + "Z"
 
@@ -76,47 +72,39 @@ def collect(data: str) -> dict:
         if buf_size >= FLUSH_SIZE:
             try:
                 n = flush_buffer()
-                return {"status": "ok", "flushed": n}
+                return json.dumps({"status": "ok", "flushed": n})
             except Exception as e:
-                return {"status": "ok", "buffered": buf_size, "flush_error": str(e)}
+                return json.dumps({"status": "ok", "buffered": buf_size, "error": str(e)})
 
-        return {"status": "ok", "buffered": buf_size}
-    except json.JSONDecodeError:
-        return {"status": "error", "message": "invalid JSON"}
+        return json.dumps({"status": "ok", "buffered": buf_size})
+    except Exception as e:
+        return json.dumps({"status": "error", "message": str(e)})
 
 
-def status():
+def get_status() -> str:
     with BUFFER_LOCK:
         buf = len(BUFFER)
-    return f"Buffer: {buf} entries | Dataset: {DATASET_REPO} | Token: {'set' if HF_TOKEN else 'MISSING'}"
+    return f"Buffer: {buf} | Token: {'✅' if HF_TOKEN else '❌'} | Dataset: {DATASET_REPO}"
 
 
-def manual_flush():
+def do_flush() -> str:
     try:
         n = flush_buffer()
-        return f"Flushed {n} entries to {DATASET_REPO}"
+        return f"Flushed {n} entries"
     except Exception as e:
-        return f"Flush error: {e}"
+        return f"Error: {e}"
 
 
 with gr.Blocks(title="肉糖生 Telemetry") as demo:
-    gr.Markdown("# 📊 肉糖生 Telemetry\nCollects Q&A from WebGPU Space → HuggingFace Dataset")
-
+    gr.Markdown("# 📊 肉糖生 Telemetry")
+    
     with gr.Row():
-        status_btn = gr.Button("Status")
-        flush_btn = gr.Button("Flush Now")
-    output = gr.Textbox(label="Result")
-
-    status_btn.click(fn=status, outputs=output)
-    flush_btn.click(fn=manual_flush, outputs=output)
-
-    # API endpoint for the WebGPU Space
-    collect_input = gr.Textbox(visible=False)
-    collect_output = gr.JSON(visible=False)
-    collect_btn = gr.Button(visible=False)
-    collect_btn.click(fn=collect, inputs=collect_input, outputs=collect_output, api_name="collect")
+        gr.Button("Status").click(fn=get_status, outputs=gr.Textbox(label="Status"))
+        gr.Button("Flush").click(fn=do_flush, outputs=gr.Textbox(label="Result"))
 
-demo.queue()
+    # Public API endpoint
+    inp = gr.Textbox(label="JSON data")
+    out = gr.Textbox(label="Response")
+    inp.submit(fn=collect, inputs=inp, outputs=out, api_name="collect")
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()