Update app.py

app.py

@@ -1,9 +1,10 @@
+import requests
 from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
-import requests
+import uvicorn
 
-# Hugging Face Space API (replace with your model space URL if different)
-HF_SPACE_API = "https://v0idalism-blacklight-text.hf.space/run/predict"
+# Hugging Face Space API to call
+HF_SPACE_API = "https://v0idalism-blacklight-image.hf.space/run/predict"
 
 # System prompt
 SYSTEM_PROMPT = (
@@ -11,9 +12,10 @@ SYSTEM_PROMPT = (
     "Respond in a brutalist, minimal, and precise style."
 )
 
+# Create app
 app = FastAPI()
 
-# Allow frontend calls
+# Allow all origins (for your site frontend)
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -23,7 +25,7 @@ app.add_middleware(
 )
 
 @app.get("/")
-def health_check():
+def root():
     return {"status": "ok", "message": "BLACKLIGHT API running"}
 
 @app.post("/predict")
@@ -34,26 +36,31 @@ async def predict(request: Request):
     if not user_prompt:
         return {"error": "No prompt provided"}
 
-    # Merge with system prompt
+    # Merge system + user prompt
     full_prompt = f"{SYSTEM_PROMPT}\n\nUser: {user_prompt}"
 
     try:
-        response = requests.post(
+        # Call Hugging Face Space without tokens
+        resp = requests.post(
             HF_SPACE_API,
             json={"data": [full_prompt]},
             timeout=60
         )
-        response.raise_for_status()
-        result = response.json()
+        resp.raise_for_status()
+        result = resp.json()
     except Exception as e:
-        return {"error": f"Failed to reach HF Space: {str(e)}"}
+        return {"error": f"Failed to contact HF Space: {str(e)}"}
 
-    # Extract first string result
+    # Extract output
     try:
         output_text = result["data"][0]
         if isinstance(output_text, list):
             output_text = output_text[0]
-    except (KeyError, IndexError, TypeError):
+    except Exception:
         return {"error": "Unexpected response format", "raw": result}
 
-    return {"reply": output_text}
+    return {"reply": output_text}
+
+# Hugging Face needs uvicorn.run in __main__
+if __name__ == "__main__":
+    uvicorn.run(app, host="0.0.0.0", port=7860)