Update app.py

app.py

@@ -10,7 +10,7 @@ import base64
 import io
 
 # ---------------------------------------------------------------------
-# Configuration - FIXED TRAILING SPACES AND URLS
+# Configuration - FIXED TRAILING SPACES
 # ---------------------------------------------------------------------
 BYTEZ_CHAT_URL   = "https://api.bytez.com/models/v2/openai/v1/chat/completions"
 BYTEZ_MODELS_URL = "https://api.bytez.com/models/v2/list/models"
@@ -63,7 +63,7 @@ async def models(authorization: str = Header(None)):
     )
 
 # ---------------------------------------------------------------------
-# /v1/chat/completions - CORRECT STREAMING
+# /v1/chat/completions - FIXED STREAMING AND SYNTAX
 # ---------------------------------------------------------------------
 @api.post("/v1/chat/completions")
 async def chat(request: Request, authorization: str = Header(None)):
@@ -85,13 +85,13 @@ async def chat(request: Request, authorization: str = Header(None)):
                     line = line.strip()
                     if not line:
                         continue
-                    if line.startswith(""):
+                    if line.startswith("data: "):
                         json_str = line[6:].strip()
                     else:
                         json_str = line
                     
                     if json_str == "[DONE]":
-                        yield " [DONE]\n\n"
+                        yield "data: [DONE]\n\n"
                         break
                     
                     try:
@@ -131,8 +131,8 @@ async def chat(request: Request, authorization: str = Header(None)):
                             "finish_reason": None,
                         }],
                     }
-                    yield f" {json.dumps(openai_chunk)}\n\n"
-        yield " [DONE]\n\n"
+                    yield f"data: {json.dumps(openai_chunk)}\n\n"
+        yield "data: [DONE]\n\n"
 
     if stream:
         return StreamingResponse(
@@ -141,7 +141,7 @@ async def chat(request: Request, authorization: str = Header(None)):
             headers={"Access-Control-Allow-Origin": "*"}
         )
     
-    # Non-streaming path
+    # Non-streaming path - FIXED SYNTAX ERROR HERE
     async with httpx.AsyncClient(timeout=120) as c:
         r = await c.post(BYTEZ_CHAT_URL, headers=headers, json=payload)
     try:
@@ -149,7 +149,8 @@ async def chat(request: Request, authorization: str = Header(None)):
     except json.JSONDecodeError:
         raise HTTPException(status_code=502, detail="Upstream returned invalid JSON")
     
-    if "choices" not in 
+    # FIXED: Complete the condition
+    if "choices" not in data:
         content = (
             data.get("output")
             or data.get("response")
@@ -166,33 +167,29 @@ async def chat(request: Request, authorization: str = Header(None)):
     return JSONResponse(data, headers={"Access-Control-Allow-Origin": "*"})
 
 # ---------------------------------------------------------------------
-# TTS ENDPOINTS (BOTH STANDARD AND NON-STANDARD)
+# TTS ENDPOINTS (ALL VARIATIONS)
 # ---------------------------------------------------------------------
 @api.post("/v1/audio/speech")
 @api.post("/v1/tts/audio/speech")
 @api.post("/v1/tts")
-@api.post("/v1/tts/audio/transcriptions")  # For backward compatibility
+@api.post("/v1/tts/audio/transcriptions")
 async def tts_endpoint(
-    request: Request,
+    request: Request = None,
     authorization: str = Header(None),
-    text: str = Form(None),  # For form-data requests
-    file: UploadFile = File(None)  # For file uploads (though TTS doesn't need files)
+    text: str = Form(None),
+    file: UploadFile = File(None)
 ):
     check_key(authorization)
     if not BYTEZ_AUTH:
         raise HTTPException(status_code=500, detail="Server BYTEZ_API_KEY not configured")
     
-    # Handle JSON payload
-    if request.headers.get("Content-Type", "").startswith("application/json"):
+    # Handle different content types
+    if request and request.headers.get("Content-Type", "").startswith("application/json"):
         try:
             payload = await request.json()
             text = payload.get("input") or payload.get("text", "")
         except:
             text = ""
-    # Handle form data
-    elif text is None and file is not None:
-        # This shouldn't happen for TTS but handle gracefully
-        text = ""
     
     if not text:
         raise HTTPException(status_code=400, detail="Missing text for TTS conversion")
@@ -216,19 +213,18 @@ async def tts_endpoint(
     # Return raw audio
     return Response(
         content=r.content,
-        media_type="audio/mpeg",  # Bytez returns MP3
+        media_type="audio/mpeg",
         headers={"Access-Control-Allow-Origin": "*"}
     )
 
 # ---------------------------------------------------------------------
-# STT ENDPOINTS (BOTH STANDARD AND NON-STANDARD)
+# STT ENDPOINTS (ALL VARIATIONS)
 # ---------------------------------------------------------------------
 @api.post("/v1/audio/transcriptions")
 @api.post("/v1/stt")
 @api.post("/v1/stt/audio/transcriptions")
-@api.post("/v1/audio/voices")  # Fake endpoint for compatibility
-@api.get("/v1/audio/voices")   # Fake GET endpoint
-@api.get("/v1/stt/audio/models")  # Fake endpoint
+@api.get("/v1/audio/voices")
+@api.get("/v1/stt/audio/models")
 async def stt_endpoint(
     request: Request = None,
     authorization: str = Header(None),
@@ -237,13 +233,12 @@ async def stt_endpoint(
 ):
     # Handle fake endpoints that just return empty responses
     if request and request.url.path in ["/v1/audio/voices", "/v1/stt/audio/models"]:
-        return JSONResponse({"voices": [], "models": []})
+        return JSONResponse({"data": []})
     
     check_key(authorization)
     if not BYTEZ_AUTH:
         raise HTTPException(status_code=500, detail="Server BYTEZ_API_KEY not configured")
     
-    # Handle file upload
     if file is None:
         raise HTTPException(status_code=400, detail="No audio file provided")
     
@@ -279,7 +274,6 @@ async def stt_endpoint(
     try:
         data = r.json()
     except json.JSONDecodeError:
-        # If response isn't JSON, try to extract text from raw response
         transcript = r.text.strip()
         if not transcript:
             raise HTTPException(status_code=502, detail="STT returned empty response")
@@ -305,7 +299,7 @@ with gr.Blocks() as ui:
                 "- `/v1/chat/completions`\n"
                 "- `/v1/audio/speech` (TTS)\n"
                 "- `/v1/audio/transcriptions` (STT)\n\n"
-                "**Also compatible with non-standard endpoints used by some clients**")
+                "**Also compatible with non-standard endpoints**")
 
 demo = gr.mount_gradio_app(api, ui, path="/")