Update main.py

main.py

@@ -372,6 +372,7 @@ async def search_gpt(q: str, stream: Optional[bool] = False, systemprompt: Optio
         return JSONResponse(content={"response": collected_text})
 
 # Enhanced streaming with direct SSE pass-through for real-time responses
+header_url = os.getenv('HEADER_URL')
 @app.post("/chat/completions")
 @app.post("/api/v1/chat/completions")
 async def get_completion(payload: Payload, request: Request, authenticated: bool = Depends(verify_api_key)):
@@ -420,9 +421,9 @@ async def get_completion(payload: Payload, request: Request, authenticated: bool
     else:
         endpoint = env_vars['secret_api_endpoint']
         custom_headers = { 
-            "Origin": "https://gpt-chatbotru-4-o1.ru",
+            "Origin": header_url,
             "Priority": "u=1, i",
-            "Referer": "https://gpt-chatbotru-4-o1.ru/"
+            "Referer": header_url
         }
     
     print(f"Using endpoint: {endpoint} for model: {model_to_use}")