Update main.py

main.py

@@ -143,9 +143,10 @@ async def get_models():
 async def fetch_models():
     return await get_models()
 
-server_status = False #working
+ENDPOINT_ORIGIN = os.getenv('ENDPOINT_ORIGIN', 'https://default-endpoint.com')
+
 @app.post("/chat/completions")
-@app.post("/v1/chat/completions")
+@app.post("api/v1/chat/completions")
 async def get_completion(payload: Payload, request: Request):
     # Check server status
     if not server_status:
@@ -162,13 +163,17 @@ async def get_completion(payload: Payload, request: Request):
             status_code=400,
             detail=f"Model '{model_to_use}' is not available. Check /models for the available model list."
         )
+    
     usage_tracker.record_request(model=model_to_use, endpoint="/chat/completions")
-    # Proceed with the request handling
+    
+    # Prepare payload
     payload_dict = payload.dict()
     payload_dict["model"] = model_to_use
     
     # Select the appropriate endpoint
     endpoint = secret_api_endpoint_2 if model_to_use in alternate_models else secret_api_endpoint
+    
+    # Current time and IP logging
     current_time = (datetime.datetime.utcnow() + datetime.timedelta(hours=5, minutes=30)).strftime("%Y-%m-%d %I:%M:%S %p")
     aaip = request.client.host
     print(f"Time: {current_time}, {aaip}")
@@ -176,11 +181,25 @@ async def get_completion(payload: Payload, request: Request):
     
     async def stream_generator(payload_dict):
         scraper = cloudscraper.create_scraper()  # Create a CloudScraper session
+        
+        # Prepare custom headers
+        custom_headers = {
+            'DNT': '1',
+            'Origin': ENDPOINT_ORIGIN,
+            'Priority': 'u=1, i',
+            'Referer': ENDPOINT_ORIGIN
+        }
+        
         try:
-            # Send POST request using CloudScraper
-            response = scraper.post(f"{endpoint}/v1/chat/completions", json=payload_dict, stream=True)
+            # Send POST request using CloudScraper with custom headers
+            response = scraper.post(
+                f"{endpoint}/v1/chat/completions", 
+                json=payload_dict, 
+                headers=custom_headers,
+                stream=True
+            )
             
-            # Check response status
+            # Error handling remains the same as in previous version
             if response.status_code == 422:
                 raise HTTPException(status_code=422, detail="Unprocessable entity. Check your payload.")
             elif response.status_code == 400: