Update app.py

app.py

@@ -1,19 +1,22 @@
-from flask import Flask, request, render_template
+from fastapi import FastAPI, Form, Request
+from fastapi.responses import HTMLResponse
+from fastapi.templating import Jinja2Templates
 
-app = Flask(__name__)
+app = FastAPI()
+templates = Jinja2Templates(directory="templates")
 
-@app.route('/')
-def index():
-    return render_template('index.html')
+@app.get("/", response_class=HTMLResponse)
+async def read_form(request: Request):
+    return templates.TemplateResponse("index.html", {"request": request})
 
-@app.route('/predict', methods=['POST'])
-def predict():
+@app.post("/predict")
+async def predict(input_text: str = Form(...)):
     # Example of processing input
-    input_text = request.form['input_text']
-    # Perform some NLP tasks here (e.g., using transformers library)
     output_text = "This is the output based on input: " + input_text
-    return render_template('index.html', output=output_text)
+    return {"output": output_text}
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)
 
-if __name__ == '__main__':
-    app.run(debug=True)