Create app.py

app.py

@@ -0,0 +1,52 @@
+import re
+import requests
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+from dotenv import load_dotenv
+import os
+
+# Load environment variables
+load_dotenv()
+
+# Get environment variables
+API_URL = os.getenv("API_URL")
+MODEL = os.getenv("MODEL")
+Origin = os.getenv("ORIGIN")
+app = FastAPI()
+
+class RoastRequest(BaseModel):
+    content: str
+
+def fetch_roasts(content: str):
+    try:
+        # Make API request
+        response = requests.post(
+            API_URL,
+            headers={'origin': Origin},
+            data={'content': content, 'model': MODEL}
+        )
+        response.raise_for_status()  # Raises HTTPError for bad responses
+
+        # Parse the result and extract roasts
+        result = response.json().get('result', '')
+        roasts = re.findall(r'<roast>(.*?)</roast>', result)
+        return roasts
+
+    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.post("/generate-roasts/")
+async def generate_roasts(request: RoastRequest):
+    roasts = fetch_roasts(request.content)
+    if not roasts:
+        raise HTTPException(status_code=404, detail="No roasts found.")
+    return {"roasts": roasts}
+
+if __name__ == "__main__":
+    import uvicorn
+    print("Starting FastAPI server...")
+    uvicorn.run(app, host="0.0.0.0", port=8000)