Update app.py

app.py

@@ -1,25 +1,26 @@
-# app.py
-#from flask import Flask, render_template, request, jsonify
-
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
+from fastapi.responses import HTMLResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
-from fastapi.responses import FileResponse
-
 from transformers import pipeline
 
 app = FastAPI()
-#model = pipeline('question-answering', model='bert-large-uncased-whole-word-masking-finetuned-squad', revision='main')
 
-@app.route('/')
-def home():
-    return render_template('index.html')
+# Uncomment and set up the model if needed
+# model = pipeline('question-answering', model='bert-large-uncased-whole-word-masking-finetuned-squad', revision='main')
+
+# Serve static files (e.g., CSS, JS, images)
+app.mount("/static", StaticFiles(directory="static"), name="static")
 
-@app.route('/predict', methods=['POST'])
-def predict():
-    user_input = request.json.get('message')
-    #result = model(user_input)
-    #response_text = result[0]['generated_text']
-    return jsonify({'response': "Hello, Hugging Face!"})#response_text
+@app.get("/", response_class=HTMLResponse)
+async def home(request: Request):
+    with open("templates/index.html") as f:
+        html_content = f.read()
+    return HTMLResponse(content=html_content)
 
-#if __name__ == '__main__':
-#    app.run(host='0.0.0.0', port=5000)
+@app.post("/predict", response_class=JSONResponse)
+async def predict(request: Request):
+    data = await request.json()
+    user_input = data.get('message')
+    # result = model(user_input)
+    # response_text = result[0]['generated_text']
+    return JSONResponse(content={'response': "Hello, Hugging Face!"})  # response_text