Update app.py

app.py

@@ -1,57 +1,43 @@
-import os
-from fastapi import FastAPI
-import uvicorn
+from fastapi import FastAPI, Request
+from fastapi.middleware.cors import CORSMiddleware
 from sentence_transformers import SentenceTransformer
-from threading import Thread
-from typing import Optional
-
-# Initialize FastAPI
-app = FastAPI(title="Sentence Transformer API")
+import uvicorn
 
-# Global model variable
-model: Optional[SentenceTransformer] = None
+# Initialize with root_path (MANDATORY FOR SPACES)
+app = FastAPI(root_path="/proxy")  # 👈 THIS IS CRITICAL
 
-def load_model():
-    """Background thread for model loading"""
-    global model
-    try:
-        model = SentenceTransformer('bge-small-zh')
-        print("Model loaded successfully")
-    except Exception as e:
-        print(f"Model loading failed: {str(e)}")
+# CORS (required for Spaces)
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
 
-# Start model loading immediately
-Thread(target=load_model, daemon=True).start()
+# Load model (simplified)
+model = SentenceTransformer('all-MiniLM-L6-v2')
 
 @app.get("/")
-def health_check():
-    """Required health check endpoint"""
+def home():
     return {
-        "status": "ready",
-        "model_loaded": model is not None
+        "message": "API is working!",
+        "endpoints": {
+            "embed": "POST /embed",
+            "docs": "/docs"
+        }
     }
 
 @app.post("/embed")
-async def embed_text(text: str):
-    if model is None:
-        return {"error": "Model still loading"}, 503
-    
-    # Convert text to list if single string
-    input_text = [text] if isinstance(text, str) else text
-    
-    embeddings = model.encode(input_text)
-    
+async def embed_text(request: Request):
+    data = await request.json()
     return {
-        "text": text,
-        "embedding": embeddings.tolist(),
-        "dimension": len(embeddings[0]) if isinstance(embeddings, list) else len(embeddings)
+        "embedding": model.encode(data['text']).tolist()
     }
 
 if __name__ == "__main__":
-    port = int(os.environ.get("PORT", 7860)) 
     uvicorn.run(
-        "app:app",
+        app,
         host="0.0.0.0",
-        port=port,
-        reload=False  # Disable auto-reload in production
+        port=7860,
+        reload=False
     )