Update app/main.py

app/main.py

@@ -1,57 +1,74 @@
 """
-CryptoSentinel AI – FastAPI entry‑point
+CryptoSentinel AI — FastAPI app entry point
+- Fetches live prices from CoinGecko
+- Provides real-time sentiment analysis using Hugging Face Transformers
+- Compatible with Hugging Face Spaces (safe caching)
 """
+
 import os
+from pathlib import Path
+
 from fastapi import FastAPI, Request, BackgroundTasks
-from fastapi.responses import HTMLResponse, StreamingResponse, JSONResponse
+from fastapi.responses import HTMLResponse, JSONResponse, StreamingResponse
 from fastapi.staticfiles import StaticFiles
+from fastapi.templating import Jinja2Templates
 from apscheduler.schedulers.background import BackgroundScheduler
-from dotenv import load_dotenv
+
 from price_fetcher import fetch_prices, CURRENT_PRICES
-from sentiment import analyze_sentiment, SentimentCache
-from pathlib import Path
+from sentiment import SentimentCache
 import json
 import asyncio
 
-load_dotenv()
+# ──────────────────────────────────────────────────────────────────────────────
 
-app = FastAPI(title="CryptoSentinel AI")
-templ_dir = Path(__file__).parent / "templates"
-app.mount("/static", StaticFiles(directory=Path(__file__).parent / "static"), name="static")
+# Define paths
+BASE_DIR = Path(__file__).parent
+TEMPLATES_DIR = BASE_DIR / "templates"
+STATIC_DIR = BASE_DIR / "static"
 
+# Create FastAPI app
+app = FastAPI(title="CryptoSentinel AI")
+templates = Jinja2Templates(directory=str(TEMPLATES_DIR))
+app.mount("/static", StaticFiles(directory=STATIC_DIR), name="static")
+
+# ──────────────────────────────────────────────────────────────────────────────
+
+# Start background job to update prices
 scheduler = BackgroundScheduler(daemon=True)
-scheduler.add_job(fetch_prices, "interval", seconds=10)  # refresh price cache
+scheduler.add_job(fetch_prices, trigger="interval", seconds=10)
 scheduler.start()
 
 @app.on_event("shutdown")
-def shutdown_event():
+def stop_scheduler():
     scheduler.shutdown(wait=False)
 
-# ---------- ROUTES -----------------------------------------------------------
+@app.on_event("startup")
+def warm_up_model():
+    # Pre-warm Hugging Face model on startup to avoid first-request latency
+    SentimentCache.compute("The market is bullish today!")
+
+# ──────────────────────────────────────────────────────────────────────────────
+# ROUTES
 
 @app.get("/", response_class=HTMLResponse)
 async def index(request: Request):
-    from fastapi.templating import Jinja2Templates
-    templates = Jinja2Templates(directory=str(templ_dir))
     return templates.TemplateResponse("index.html", {"request": request})
 
-@app.get("/prices")
-async def prices():
-    """JSON endpoint for latest cached prices."""
-    return JSONResponse(CURRENT_PRICES)
+@app.get("/prices", response_class=JSONResponse)
+async def get_prices():
+    return CURRENT_PRICES
 
 @app.post("/sentiment")
-async def sentiment(request: Request, background_tasks: BackgroundTasks):
+async def analyze_sentiment(request: Request, background_tasks: BackgroundTasks):
     body = await request.json()
     text = body.get("text", "")
-    # Run expensive inference in background to keep latency low
     background_tasks.add_task(SentimentCache.compute, text)
     return {"status": "processing"}
 
 @app.get("/sentiment/stream")
-async def sentiment_stream():
-    """Server‑sent events stream – pushes sentiment results."""
-    async def event_generator():
+async def stream_sentiment():
+    """Server-Sent Events stream for pushing sentiment updates."""
+    async def event_stream():
         last_id = 0
         while True:
             if SentimentCache.latest_id != last_id:
@@ -59,4 +76,5 @@ async def sentiment_stream():
                 data = json.dumps(SentimentCache.latest_result)
                 yield f"id:{last_id}\ndata:{data}\n\n"
             await asyncio.sleep(1)
-    return StreamingResponse(event_generator(), media_type="text/event-stream")
+
+    return StreamingResponse(event_stream(), media_type="text/event-stream")