Update app.py

app.py

@@ -2,6 +2,8 @@ from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import JSONResponse
 from fastapi.staticfiles import StaticFiles
+from fastapi import Request
+import requests
 import numpy as np
 import argparse
 import os
@@ -25,6 +27,7 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+HF_TOKEN = os.getenv("REACT_APP_HF_TOKEN")
 
 @app.get("/invert")
 async def invert(text: str):
@@ -33,6 +36,26 @@ async def invert(text: str):
         "inverted": text[::-1],
     }
 
+@app.post("/api/chat")
+async def chat(request: Request):
+    data = await request.json()
+    user_message = data.get("message", "")
+    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+    payload = {
+        "inputs": {
+            "past_user_inputs": [],
+            "generated_responses": [],
+            "text": user_message
+        }
+    }
+    response = requests.post(
+        "https://api-inference.huggingface.co/models/mistralai/Mistral-7B-Instruct-v0.3",
+        headers=headers,
+        json=payload
+    )
+    result = response.json()
+    return {"response": result}
+
 
 @app.get("/data")
 async def get_data():