Spaces:

userIdc2024
/

Creative_Breakthrough

Running

sushilideaclan01 commited on Jan 14

Commit

a18335a

1 Parent(s): 8454aa1

Fix: Use Response instead of StreamingResponse for better HF compatibility

Files changed (1) hide show

main.py CHANGED Viewed

@@ -334,16 +334,20 @@ async def root():
     try:
         async with httpx.AsyncClient(timeout=30.0) as client:
             response = await client.get("http://localhost:3000/")
-            return StreamingResponse(
-                response.iter_bytes(),
                 status_code=response.status_code,
-                headers=dict(response.headers),
                 media_type=response.headers.get("content-type"),
             )
     except httpx.RequestError:
-        raise HTTPException(
-            status_code=503,
-            detail="Frontend server is not available."
         )

     try:
         async with httpx.AsyncClient(timeout=30.0) as client:
             response = await client.get("http://localhost:3000/")
+            # Return full response (not streaming) for better HF compatibility
+            from fastapi.responses import Response
+            return Response(
+                content=response.content,
                 status_code=response.status_code,
+                headers={k: v for k, v in response.headers.items() if k.lower() not in ['content-encoding', 'transfer-encoding', 'content-length']},
                 media_type=response.headers.get("content-type"),
             )
     except httpx.RequestError:
+        # Return a simple HTML page if frontend is not ready yet
+        return Response(
+            content="<html><head><meta http-equiv='refresh' content='2'></head><body><h1>Loading...</h1></body></html>",
+            status_code=200,
+            media_type="text/html"
         )