Update app.py

app.py

@@ -56,10 +56,19 @@ def gen(prompt):
         raise HTTPException(status_code=500, detail="AI Generation Failed")
 
 
+class VoteInput(BaseModel):
+    unique_id: str
+    is_upvote: bool
+
+
 @app.post("/vote")
-async def vote(unique_id: str, is_upvote: bool):
-    data = {"id": unique_id, "vote": str(is_upvote)}
-    response = requests.post(f"{Url}/vote", json=data)
+async def vote(vote_input: VoteInput):
+    data = {"id": vote_input.unique_id, "vote": str(vote_input.is_upvote)}
+    try:
+        response = requests.post(f"{Url}/vote", json=data, timeout=10)
+        response.raise_for_status()
+    except requests.RequestException as e:
+        raise HTTPException(status_code=502, detail=f"Failed to record vote: {str(e)}")
     return {"status": "success"}