initial

Dockerfile

@@ -24,4 +24,4 @@ ENV NAME World
 # Use gunicorn for a more robust production server (optional, but recommended)
 # CMD ["gunicorn", "--bind", "0.0.0.0:7860", "app:app"]
 # Fallback to Flask's built-in server for simplicity
-CMD ["python", "app.py"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -1,14 +1,16 @@
-from flask import Flask, request, jsonify
+from fastapi import FastAPI, Request, HTTPException
+from fastapi.responses import JSONResponse
 import logging
 import sys
+import uvicorn  # Import uvicorn
 
 # Configure logging to output to stdout
 logging.basicConfig(stream=sys.stdout, level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-app = Flask(__name__)
+app = FastAPI()
 
-@app.route('/', methods=['POST'])
-def handle_request():
+@app.post("/")
+async def handle_request(request: Request):
     """Handles incoming POST requests, logs details, and returns a response."""
     logging.info("="*20 + " New Request Received " + "="*20)
 
@@ -17,27 +19,41 @@ def handle_request():
     for header, value in request.headers.items():
         logging.info(f"  {header}: {value}")
 
-    # Log JSON body if present
-    content_type = request.headers.get('Content-Type')
-    if content_type and 'application/json' in content_type:
-        try:
-            json_data = request.get_json()
+    # Log body
+    content_type = request.headers.get('content-type')
+    logging.info("Body:")
+    try:
+        if content_type and 'application/json' in content_type:
+            json_data = await request.json()
             logging.info("JSON Body:")
             logging.info(json_data)
-        except Exception as e:
-            logging.error(f"Error parsing JSON: {e}")
-            logging.info("Raw Body (JSON parsing failed):")
-            logging.info(request.get_data(as_text=True))
-    else:
-        # Log raw body if not JSON
-        logging.info("Raw Body:")
-        logging.info(request.get_data(as_text=True))
+        else:
+            body_data = await request.body()
+            logging.info("Raw Body:")
+            # Try decoding as text, fallback to raw bytes if fails
+            try:
+                logging.info(body_data.decode('utf-8'))
+            except UnicodeDecodeError:
+                 logging.info(f"Raw Body (non-UTF-8): {body_data}")
+    except Exception as e:
+        logging.error(f"Error processing body: {e}")
+        # Log raw body on error as well
+        try:
+            body_data = await request.body()
+            logging.info(f"Raw Body (on error): {body_data}")
+        except Exception as read_err:
+             logging.error(f"Could not even read raw body on error: {read_err}")
+        raise HTTPException(status_code=400, detail=f"Error processing request body: {e}")
+
 
     logging.info("="*20 + " End of Request " + "="*20 + "\n")
 
-    return jsonify({"status": "success", "message": "Request received and logged."}), 200
+    return JSONResponse(content={"status": "success", "message": "Request received and logged."}, status_code=200)
 
-if __name__ == '__main__':
-    # Hugging Face Spaces typically expect the app on port 7860
-    # Using 0.0.0.0 to be accessible externally within the container
-    app.run(host='0.0.0.0', port=7860)
+# The following block is typically not needed when running with Uvicorn via CLI,
+# but can be useful for direct execution (e.g., local debugging).
+# Hugging Face Spaces will use the CMD in the Dockerfile.
+# if __name__ == "__main__":
+#     # Default Uvicorn port is 8000, suitable for many environments.
+#     # Use 0.0.0.0 to be accessible externally.
+#     uvicorn.run(app, host="0.0.0.0", port=8000)

requirements.txt

@@ -1 +1,2 @@
-Flask==3.0.3
+fastapi
+uvicorn[standard]