Update app.py

app.py

@@ -6,6 +6,7 @@ from dotenv import load_dotenv
 import os
 from fake_useragent import UserAgent
 from fastapi.responses import RedirectResponse
+
 # Load environment variables
 load_dotenv()
 
@@ -38,11 +39,12 @@ def fetch_roasts(content: str):
     }
 
     try:
-        # Make API request
+        # Make API request with a 30-second timeout
         response = requests.post(
             API_URL,
             headers=headers,
-            data={'content': content, 'model': MODEL}
+            data={'content': content, 'model': MODEL},
+            timeout=30  # Add timeout here
         )
         response.raise_for_status()  # Raises HTTPError for bad responses
 
@@ -51,17 +53,19 @@ def fetch_roasts(content: str):
         roasts = re.findall(r'<roast>(.*?)</roast>', result)
         return roasts
 
+    except requests.exceptions.Timeout:
+        raise HTTPException(status_code=504, detail="The request timed out.")
     except requests.exceptions.RequestException as e:
         raise HTTPException(status_code=503, detail=f"Network error: {e}")
     except KeyError as e:
         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)
@@ -73,4 +77,3 @@ if __name__ == "__main__":
     import uvicorn
     print("Starting FastAPI server...")
     uvicorn.run(app, host="0.0.0.0", port=8000)
-