Update proxy_server.py

proxy_server.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI, HTTPException
-from fastapi.staticfiles import StaticFiles  # ADD THIS
-from fastapi.responses import FileResponse    # ADD THIS
+from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
 import httpx
 import os
 from pydantic import BaseModel
@@ -8,7 +8,6 @@ import uvicorn
 
 app = FastAPI()
 
-# Pydantic model for earthquake request
 class EarthquakeRequest(BaseModel):
     start_date: str
     end_date: str
@@ -17,44 +16,32 @@ class EarthquakeRequest(BaseModel):
 async def proxy_to_lambda(earthquake_request: EarthquakeRequest):
     """Proxy requests to your API Gateway Lambda"""
     try:
-        # Get API key from environment variables
         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")
-        
-        # Your actual API Gateway URL
-        api_url = os.environ.get('API_GATEWAY_URL')
-        
         if not api_url:
             raise HTTPException(status_code=500, detail="API Gateway URL not configured")
         
-        # Headers to send to your API Gateway
         headers = {
             'Content-Type': 'application/json',
             'X-API-Key': api_key
         }
         
-        # Request body
         request_body = {
             'start_date': earthquake_request.start_date,
             'end_date': earthquake_request.end_date
         }
         
-        # Make request to your API Gateway
         async with httpx.AsyncClient(timeout=30.0) as client:
-            response = await client.post(
-                api_url,
-                json=request_body,
-                headers=headers
-            )
+            response = await client.post(api_url, json=request_body, headers=headers)
             
-            # Return the response from your API Gateway
             if response.status_code == 200:
                 return response.json()
             else:
                 raise HTTPException(
-                    status_code=response.status_code, 
+                    status_code=response.status_code,
                     detail=f"API Gateway error: {response.text}"
                 )
                 
@@ -63,16 +50,15 @@ async def proxy_to_lambda(earthquake_request: EarthquakeRequest):
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Proxy error: {str(e)}")
 
-# Health check endpoint
 @app.get("/api/health")
 async def health_check():
-    return {"status": "healthy", "service": "earthquake-proxy"}
+    return {"status": "healthy"}
 
-# Mount static files from dist directory
+# Mount static assets
 if os.path.exists("dist/assets"):
     app.mount("/assets", StaticFiles(directory="dist/assets"), name="assets")
 
-# Serve React app for all non-API routes
+# Serve React app
 @app.get("/{full_path:path}")
 async def serve_react_app(full_path: str = ""):
     if full_path.startswith("api/"):