convert to flask + cors

app.py

@@ -1,45 +1,48 @@
-from typing import Dict
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
+from flask import Flask, request, jsonify, abort
 import os
 import httpx
+from flask_cors import CORS
 
 # Load Mistral API token from environment variables
 MISTRAL_API_TOKEN = os.getenv("MISTRAL_API_TOKEN")
 if not MISTRAL_API_TOKEN:
     raise EnvironmentError("MISTRAL_API_TOKEN environment variable not set.")
 
-# Initialize FastAPI app
-app = FastAPI()
+# Initialize Flask app
+app = Flask(__name__)
+CORS(app)
 
-# # Define the request model for input validation
-# class MistralRequest(BaseModel):
-#     system_prompt : str
-#     input_text: str
-
-@app.post("/proxy/mistral")
-async def proxy_mistral(request_data: Dict):
+@app.route("/proxy/mistral", methods=["POST"])
+def proxy_mistral():
     """Proxy endpoint to send requests to Mistral API."""
+
+    # Mistral endpoint and headers
     url = "https://api.mistral.ai/v1/chat/completions"
     headers = {
         "Authorization": f"Bearer {MISTRAL_API_TOKEN}",
         "Content-Type": "application/json"
     }
-    
 
-    # Construct the data payload    
-    print("attempting to sendc the request")
+    # Parse the JSON body from the incoming request
+    request_data = request.get_json()
+    print("Attempting to send the request to Mistral")
     print(request_data)
 
-    # Send the request to the Mistral API
-    async with httpx.AsyncClient() as client:
-        response = await client.post(url, headers=headers, json=request_data, timeout=None)
-        print("my response" , response)
+    # Send the request to the Mistral API using httpx (synchronous)
+    with httpx.Client() as client:
+        response = client.post(url, headers=headers, json=request_data, timeout=None)
 
-    # Check the response status
+    # If Mistral doesn’t return a 200, propagate the error
     if response.status_code != 200:
-        raise HTTPException(status_code=response.status_code, detail=response.text)
-    print(response.json())
-    # Return the Mistral API response
-    return response.json()
-
+        # In Flask, you can either return an error explicitly:
+        #   return (response.text, response.status_code)
+        # or raise an HTTPException via `abort()`.
+        return (response.text, response.status_code)
+
+    # Print the Mistral API response (for debug) and return it
+    print("Mistral response:", response.json())
+    return jsonify(response.json())
+
+if __name__ == "__main__":
+    # Run the Flask app (use your desired host and port)
+    app.run(host="0.0.0.0", port=8000, debug=True)

requirements.txt

@@ -1,3 +1,4 @@
-fastapi
+flask
+flask-cors
 uvicorn[standard]
 httpx