Update app.py

app.py

@@ -1,10 +1,10 @@
 import re
 import requests
 from fastapi import FastAPI, HTTPException
-from fastapi.responses import RedirectResponse
 from pydantic import BaseModel
 from dotenv import load_dotenv
 import os
+from fake_useragent import UserAgent
 
 # Load environment variables
 load_dotenv()
@@ -27,11 +27,21 @@ class RoastRequest(BaseModel):
     content: str
 
 def fetch_roasts(content: str):
+    ua = UserAgent()
+    headers = {
+        'origin': Origin,
+        'user-agent': ua.random,
+        'accept': '*/*',
+        'accept-encoding': 'gzip, deflate, br, zstd',
+        'accept-language': 'en-US,en;q=0.9,en-IN;q=0.8',
+        'referer': 'https://seoroast.co/tools/ai-roast-generator?via=topaitools',
+    }
+
     try:
         # Make API request
         response = requests.post(
             API_URL,
-            headers={'origin': Origin},
+            headers=headers,
             data={'content': content, 'model': MODEL}
         )
         response.raise_for_status()  # Raises HTTPError for bad responses
@@ -47,11 +57,11 @@ def fetch_roasts(content: str):
         raise HTTPException(status_code=500, detail=f"Response format error: {e}")
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Unexpected error: {e}")
-
+        
 @app.get("/", include_in_schema=False)
 async def root():
     return RedirectResponse(url="https://roastapi-docs.netlify.app/")
-
+    
 @app.post("/generate-roasts/")
 async def generate_roasts(request: RoastRequest):
     roasts = fetch_roasts(request.content)
@@ -63,3 +73,4 @@ if __name__ == "__main__":
     import uvicorn
     print("Starting FastAPI server...")
     uvicorn.run(app, host="0.0.0.0", port=8000)
+