| from fastapi import FastAPI, HTTPException |
| from fastapi.staticfiles import StaticFiles |
| from fastapi.responses import FileResponse |
| import httpx |
| import os |
| from pydantic import BaseModel |
| import uvicorn |
|
|
| app = FastAPI() |
|
|
| class EarthquakeRequest(BaseModel): |
| start_date: str |
| end_date: str |
|
|
| @app.post("/api/proxy") |
| async def proxy_to_lambda(earthquake_request: EarthquakeRequest): |
| """Proxy requests to your API Gateway Lambda""" |
| try: |
| api_key = os.environ.get('API_GATEWAY_KEY') |
| api_url = os.environ.get('API_GATEWAY_URL') |
| |
| if not api_key: |
| raise HTTPException(status_code=500, detail="API key not configured") |
| if not api_url: |
| raise HTTPException(status_code=500, detail="API Gateway URL not configured") |
| |
| headers = { |
| 'Content-Type': 'application/json', |
| 'X-API-Key': api_key |
| } |
| |
| request_body = { |
| 'start_date': earthquake_request.start_date, |
| 'end_date': earthquake_request.end_date |
| } |
| |
| async with httpx.AsyncClient(timeout=30.0) as client: |
| response = await client.post(api_url, json=request_body, headers=headers) |
| |
| if response.status_code == 200: |
| return response.json() |
| else: |
| raise HTTPException( |
| status_code=response.status_code, |
| detail=f"API Gateway error: {response.text}" |
| ) |
| |
| except httpx.TimeoutException: |
| raise HTTPException(status_code=504, detail="Request timeout") |
| except Exception as e: |
| raise HTTPException(status_code=500, detail=f"Proxy error: {str(e)}") |
|
|
| @app.get("/api/health") |
| async def health_check(): |
| return {"status": "healthy"} |
|
|
| |
| if os.path.exists("dist/assets"): |
| app.mount("/assets", StaticFiles(directory="dist/assets"), name="assets") |
|
|
| |
| @app.get("/{full_path:path}") |
| async def serve_react_app(full_path: str = ""): |
| if full_path.startswith("api/"): |
| raise HTTPException(status_code=404, detail="API endpoint not found") |
| return FileResponse("dist/index.html") |
|
|
| if __name__ == "__main__": |
| uvicorn.run(app, host="0.0.0.0", port=7860) |
