Update app/app.py

app/app.py

@@ -7,14 +7,19 @@ real-time signal delivery. This is the definitive money-spinning engine.
 import asyncio
 import os
 from contextlib import asynccontextmanager
+import json
+import time # Added import
 from datetime import datetime, timezone
+
 import httpx
 import socketio
 from fastapi import FastAPI
 from fastapi.staticfiles import StaticFiles
 
-from price_fetcher import PriceFetcher
-from arbitrage_analyzer import ArbitrageAnalyzer
+# Relative imports now work perfectly because of the package structure
+from .price_fetcher import PriceFetcher
+from .arbitrage_analyzer import ArbitrageAnalyzer
+from .broker import signal_broker
 
 OPPORTUNITY_THRESHOLD = 0.001
 
@@ -26,24 +31,26 @@ socket_app = socketio.ASGIApp(sio)
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     print("🚀 Initializing Sentinel Arbitrage Engine v10.0...")
-    price_fetcher = PriceFetcher(httpx.AsyncClient())
-    arbitrage_analyzer = ArbitrageAnalyzer(httpx.AsyncClient())
-    
-    # Launch the engine as a background task, passing the Socket.IO server instance
-    arbitrage_task = asyncio.create_task(
-        run_arbitrage_detector(price_fetcher, arbitrage_analyzer)
-    )
-    
-    print("✅ Engine is online and hunting for opportunities.")
-    yield
-    
-    print("⏳ Shutting down engine...")
-    arbitrage_task.cancel()
-    try: await arbitrage_task
-    except asyncio.CancelledError: print("Engine shut down gracefully.")
+    # Use a single httpx client for the application's lifespan
+    async with httpx.AsyncClient() as client:
+        app.state.price_fetcher = PriceFetcher(client)
+        app.state.arbitrage_analyzer = ArbitrageAnalyzer(client)
+        
+        arbitrage_task = asyncio.create_task(
+            run_arbitrage_detector(app.state.price_fetcher, app.state.arbitrage_analyzer)
+        )
+        
+        print("✅ Engine is online and hunting for opportunities.")
+        yield
+        
+        print("⏳ Shutting down engine...")
+        arbitrage_task.cancel()
+        try: await arbitrage_task
+        except asyncio.CancelledError: print("Engine shut down gracefully.")
 
 async def run_arbitrage_detector(price_fetcher, analyzer):
     """The core engine loop. Now directly emits events via Socket.IO."""
+    last_opportunity_time = 0
     while True:
         try:
             await price_fetcher.update_prices_async()
@@ -55,18 +62,18 @@ async def run_arbitrage_detector(price_fetcher, analyzer):
             if pyth_price and chainlink_price:
                 spread = abs(pyth_price - chainlink_price) / chainlink_price
                 if spread > OPPORTUNITY_THRESHOLD:
-                    opportunity = {
-                        "pyth_price": pyth_price,
-                        "chainlink_price": chainlink_price,
-                        "spread_pct": spread * 100
-                    }
-                    print(f"⚡️ Discrepancy Detected: {opportunity['spread_pct']:.3f}%")
-                    briefing = await analyzer.get_alpha_briefing(opportunity)
-                    if briefing:
-                        signal = {**opportunity, **briefing, "timestamp": datetime.now(timezone.utc).isoformat()}
-                        # --- THE FIX: Directly emit to all connected clients ---
-                        await sio.emit('new_signal', signal)
-                        print(f"✅ Signal Emitted: {signal['strategy']}")
+                    current_time = time.time()
+                    if current_time - last_opportunity_time > 30: # Throttle Gemini calls
+                        last_opportunity_time = current_time
+                        opportunity = {
+                            "pyth_price": pyth_price, "chainlink_price": chainlink_price, "spread_pct": spread * 100
+                        }
+                        print(f"⚡️ Discrepancy Detected: {opportunity['spread_pct']:.3f}%")
+                        briefing = await analyzer.get_alpha_briefing(opportunity)
+                        if briefing:
+                            signal = {**opportunity, **briefing, "timestamp": datetime.now(timezone.utc).isoformat()}
+                            await sio.emit('new_signal', signal)
+                            print(f"✅ Signal Emitted: {signal['strategy']}")
         except Exception as e:
             print(f"❌ ERROR in engine loop: {e}")
 
@@ -75,9 +82,6 @@ async def run_arbitrage_detector(price_fetcher, analyzer):
 # --- FastAPI App Setup ---
 app = FastAPI(lifespan=lifespan)
 
-# This serves the index.html and any other static files (like JS or CSS)
-app.mount("/", StaticFiles(directory="static", html=True), name="static")
-
 @sio.event
 async def connect(sid, environ):
     print(f"✅ Client connected: {sid}")
@@ -86,5 +90,8 @@ async def connect(sid, environ):
 async def disconnect(sid):
     print(f"🔥 Client disconnected: {sid}")
 
-# Mount the Socket.IO app on top of the FastAPI app
-app.mount('/socket.io', socket_app)
+# Mount the Socket.IO app on a specific path
+app.mount('/socket.io', socket_app)
+
+# Serve the static files (like index.html) from the 'static' directory
+app.mount("/", StaticFiles(directory="static", html=True), name="static")