Update app.py

app.py

@@ -2,10 +2,18 @@ from fastapi import FastAPI, Request
 from fastapi.middleware.cors import CORSMiddleware
 import requests
 
-# FastAPI app (must be named `app` so HF detects it)
+# Hugging Face Space API (replace with your model space URL if different)
+HF_SPACE_API = "https://v0idalism-blacklight-text.hf.space/run/predict"
+
+# System prompt
+SYSTEM_PROMPT = (
+    "You are BLACKLIGHT, a precision AI created by v0id under AWAKEN CULT VISIONS. "
+    "Respond in a brutalist, minimal, and precise style."
+)
+
 app = FastAPI()
 
-# Allow all origins so your website frontend can call this API
+# Allow frontend calls
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -14,18 +22,10 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# SYSTEM PROMPT
-SYSTEM_PROMPT = (
-    "You are BLACKLIGHT, a precision AI created by v0id under AWAKEN CULT VISIONS. "
-    "Respond in a brutalist, minimal, and precise style."
-)
-
-# Health check route for Hugging Face Space status
 @app.get("/")
 def health_check():
     return {"status": "ok", "message": "BLACKLIGHT API running"}
 
-# Prediction endpoint
 @app.post("/predict")
 async def predict(request: Request):
     data = await request.json()
@@ -34,24 +34,26 @@ async def predict(request: Request):
     if not user_prompt:
         return {"error": "No prompt provided"}
 
-    # Merge system prompt with user input
+    # Merge with system prompt
     full_prompt = f"{SYSTEM_PROMPT}\n\nUser: {user_prompt}"
 
-    # Call the hosted HF Space directly (replace with your actual text space)
-    HF_SPACE_API = "https://v0idalism-blacklight-text.hf.space/run/predict"
-
     try:
-        r = requests.post(HF_SPACE_API, json={"data": [full_prompt]}, timeout=60)
-        r.raise_for_status()
-    except requests.RequestException as e:
+        response = requests.post(
+            HF_SPACE_API,
+            json={"data": [full_prompt]},
+            timeout=60
+        )
+        response.raise_for_status()
+        result = response.json()
+    except Exception as e:
         return {"error": f"Failed to reach HF Space: {str(e)}"}
 
-    result = r.json()
-
-    # Extract text from HF Space response
+    # Extract first string result
     try:
         output_text = result["data"][0]
+        if isinstance(output_text, list):
+            output_text = output_text[0]
     except (KeyError, IndexError, TypeError):
-        return {"error": "Unexpected response format from HF Space", "raw": result}
+        return {"error": "Unexpected response format", "raw": result}
 
     return {"reply": output_text}