Making interactive

app.py

@@ -1,31 +1,62 @@
-# app.py
+import gradio as gr
 import asyncio
-import os
-os.environ["USE_MINIMAL_LIBS"] = "true"
+from triangular_arbitrage import detector
 
-import gradio as gr
-import triangular_arbitrage.detector as detector
-
-EXCHANGE = "binanceus" # or "binanceus"
-REFRESH_INTERVAL = 30  # seconds
-
-def run_detection_sync():
-    best_opps, best_profit = asyncio.run(detector.run_detection(EXCHANGE, max_cycle=3))
-    if best_opps:
-        profit_pct = round((best_profit - 1) * 100, 5)
-        lines = [f"**{profit_pct}% opportunity found**\n"]
-        for i, opp in enumerate(best_opps):
-            side = 'buy' if opp.reversed else 'sell'
-            conn = 'with' if side == 'buy' else 'for'
-            lines.append(f"{i+1}. {side} {opp.symbol.base} {conn} {opp.symbol.quote} @ {opp.last_price:.5f}")
-        return "\n".join(lines)
-    return "No opportunity detected. Scanning..."
-
-with gr.Blocks() as demo:
-    output = gr.Textbox()
-    demo.load(run_detection_sync, outputs=output)
-    # Auto-refresh every REFRESH_INTERVAL seconds
-    timer = gr.Timer(REFRESH_INTERVAL)
-    timer.tick(run_detection_sync, outputs=output)
-
-demo.launch()
+async def run_detection_ui(exchange_name):
+    try:
+        best_opps, best_profit = await detector.run_detection(
+            exchange_name=exchange_name,
+            max_cycle=3,  # fixed triangular only
+        )
+
+        if not best_opps:
+            return "No arbitrage opportunity found."
+
+        result_lines = []
+        for ticker in best_opps:
+            result_lines.append(
+                f"{ticker.symbol} | price: {ticker.last_price:.8f}"
+            )
+
+        result_text = "\n".join(result_lines)
+        result_text += f"\n\n**Total cycle profit:** {best_profit:.6f}"
+
+        return result_text
+
+    except Exception as e:
+        return f"Error while scanning: {str(e)}"
+
+
+with gr.Blocks(title="Triangular Arbitrage Scanner") as demo:
+    gr.Markdown("# Triangular Arbitrage Scanner")
+
+    exchange = gr.Dropdown(
+        choices=[
+            "binanceus",   # safer default for Spaces
+            "kraken",
+            "bitget",
+            "bybit",
+            "okx",
+            "coinbase",
+            "kucoin",
+            "gateio",
+            "huobi",
+        ],
+        value="kraken",
+        label="Exchange",
+    )
+
+    scan_btn = gr.Button("Scan for opportunities", variant="primary")
+
+    output = gr.Markdown(
+        value="Select exchange and click **Scan**.",
+        label="Result"
+    )
+
+    scan_btn.click(
+        fn=run_detection_ui,
+        inputs=exchange,
+        outputs=output,
+    )
+
+demo.launch()