Spaces:
Sleeping
Sleeping
File size: 2,318 Bytes
46db43c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
import os, logging
from dotenv import load_dotenv
load_dotenv()
from fastapi import FastAPI
from fastapi.staticfiles import StaticFiles
from fastapi.middleware.cors import CORSMiddleware
from utils.logger import get_logger
from utils.api.rotator import APIKeyRotator
from utils.ingestion.caption import BlipCaptioner
from utils.rag.embeddings import EmbeddingClient
from utils.rag.rag import RAGStore, ensure_indexes
# ────────────────────────────── App Setup ──────────────────────────────
logger = get_logger("APP", name="studybuddy")
app = FastAPI(title="StudyBuddy RAG", version="0.1.0")
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# Serve static files (index.html, scripts.js, styles.css)
app.mount("/static", StaticFiles(directory="static"), name="static")
# In-memory job tracker (for progress queries)
app.state.jobs = {}
# ────────────────────────────── Global Clients ──────────────────────────────
# API rotators (round robin + auto failover on quota errors)
gemini_rotator = APIKeyRotator(prefix="GEMINI_API_", max_slots=5)
nvidia_rotator = APIKeyRotator(prefix="NVIDIA_API_", max_slots=5)
# Captioner + Embeddings (lazy init inside classes)
captioner = BlipCaptioner()
embedder = EmbeddingClient(model_name=os.getenv("EMBED_MODEL", "sentence-transformers/all-MiniLM-L6-v2"))
# Mongo / RAG store
try:
rag = RAGStore(mongo_uri=os.getenv("MONGO_URI"), db_name=os.getenv("MONGO_DB", "studybuddy"))
# Test the connection
rag.client.admin.command('ping')
logger.info("[APP] MongoDB connection successful")
ensure_indexes(rag)
logger.info("[APP] MongoDB indexes ensured")
except Exception as e:
logger.error(f"[APP] Failed to initialize MongoDB/RAG store: {str(e)}")
logger.error(f"[APP] MONGO_URI: {os.getenv('MONGO_URI', 'Not set')}")
logger.error(f"[APP] MONGO_DB: {os.getenv('MONGO_DB', 'studybuddy')}")
# Create a dummy RAG store for now - this will cause errors but prevents the app from crashing
rag = None
|