Update main.py

main.py

@@ -10,7 +10,7 @@ app = FastAPI()
 valid_api_keys = ['PARTH-SADARIA-NI-API-CHAWI', 'HEET-NI-CHEESEWADI-API-KEY']
 
 # Fetch the secret API endpoint from environment variables
-SECRET_API_ENDPOINT = os.getenv("SECRET_API_ENDPOINT", "Default_Endpoint_Not_Set")
+api_endpoint = os.getenv("SECRET_API_ENDPOINT")
 
 class Payload(BaseModel):
     model: str
@@ -29,11 +29,6 @@ async def get_models():
     except requests.exceptions.RequestException as e:
         raise HTTPException(status_code=500, detail=f"Request failed: {e}")
 
-@app.get("/api-endpoint")
-async def get_secret_api_endpoint():
-    # Return the secret API endpoint
-    return {"secret_api_endpoint": SECRET_API_ENDPOINT}
-
 @app.post("/v1/chat/completions")
 async def get_completion(payload: Payload, request: Request):
     api_key = request.headers.get("Authorization")
@@ -42,16 +37,13 @@ async def get_completion(payload: Payload, request: Request):
     if api_key not in valid_api_keys:
         raise HTTPException(status_code=403, detail="Forbidden: Invalid API key. Join dsc.gg/chadgang and DM @mr_leaderyt on Discord for a free API key :)")
     
-    # Set the URL for all requests
-    url = "https://gpt.tiptopuni.com/api/openai/v1/chat/completions"
-    
     # Prepare the payload for streaming
     payload_dict = {**payload.dict(), "stream": True}
 
     # Define a generator to stream the response
     def stream_generator():
         try:
-            with requests.post(url, json=payload_dict, stream=True, timeout=5) as response:
+            with requests.post(api_endpoint, json=payload_dict, stream=True, timeout=10) as response:
                 response.raise_for_status()
                 for chunk in response.iter_content(chunk_size=1024):
                     if chunk:  # Only yield non-empty chunks
@@ -59,6 +51,16 @@ async def get_completion(payload: Payload, request: Request):
         except requests.exceptions.RequestException as e:
             raise HTTPException(status_code=500, detail=f"Streaming failed: {e}")
 
+    # Send a POST request to the secret API endpoint
+    try:
+        secret_api_response = requests.post(api_endpoint, json=payload_dict, timeout=5)
+        secret_api_response.raise_for_status()  # Check if the request was successful
+        # Log or process the secret API response if needed
+        print(secret_api_response.json())  # Just an example of processing the response
+
+    except requests.exceptions.RequestException as e:
+        raise HTTPException(status_code=500, detail=f"Failed to POST to secret API endpoint: {e}")
+    
     # Return the streaming response
     return StreamingResponse(stream_generator(), media_type="application/json")
 
@@ -69,4 +71,8 @@ async def startup_event():
     print("GET /")
     print("GET /models")
     print("POST /v1/chat/completions")
-    print
+
+# Run the server with Uvicorn using the 'main' module
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)