Update app.py

app.py

@@ -1,4 +1,3 @@
-
 from fastapi import FastAPI, Form, Request
 from fastapi.responses import HTMLResponse, StreamingResponse
 import os
@@ -8,9 +7,9 @@ import random
 
 app = FastAPI()
 
-# Load environment variables
+# Load environment variables with defaults
 MODEL = "gpt-4o-mini"
-API_URL = os.getenv("API_URL")
+API_URL = os.getenv("API_URL", "https://api.openai.com/v1/chat/completions")  # Default to OpenAI API
 DISABLED = os.getenv("DISABLED") == 'True'
 OPENAI_API_KEYS = os.getenv("OPENAI_API_KEYS", "").split(",")
 NUM_THREADS = int(os.getenv("NUM_THREADS", 1))
@@ -131,6 +130,12 @@ async def chat(input: str = Form(...), top_p: float = Form(1.0), temperature: fl
     if DISABLED:
         return StreamingResponse(iter(["Usage limit reached."]), media_type="text/plain")
 
+    if not API_URL:
+        return StreamingResponse(iter(["Error: API_URL is not set in the environment."]), media_type="text/plain")
+
+    if not OPENAI_API_KEYS or OPENAI_API_KEYS == [""]:
+        return StreamingResponse(iter(["Error: No valid OPENAI_API_KEYS provided."]), media_type="text/plain")
+
     payload = {
         "model": MODEL,
         "messages": [{"role": "user", "content": input}],
@@ -159,6 +164,8 @@ async def chat(input: str = Form(...), top_p: float = Form(1.0), temperature: fl
                         chunk_json = json.loads(chunk_data[6:])
                         if "choices" in chunk_json and "delta" in chunk_json["choices"][0] and "content" in chunk_json["choices"][0]["delta"]:
                             yield chunk_json["choices"][0]["delta"]["content"]
+        except requests.exceptions.MissingSchema:
+            yield "Error: Invalid API_URL. Please provide a valid URL (e.g., https://api.openai.com/v1/chat/completions)."
         except Exception as e:
             yield f"Error: {str(e)}"