Update app.py

app.py

@@ -1,6 +1,4 @@
-from fastapi import FastAPI, Query
-from fastapi.responses import FileResponse, JSONResponse
-import uvicorn
+from flask import Flask, request, jsonify, send_from_directory
 import fitz  # PyMuPDF for PDF text extraction
 import faiss  # FAISS for vector search
 import numpy as np
@@ -25,17 +23,22 @@ index = faiss.IndexFlatL2(vector_dim)  # FAISS index
 
 documents = []  # Store extracted text
 
-app = FastAPI()
+app = Flask(__name__)
 
-@app.get("/")
+@app.route("/")
 def serve_homepage():
     """Serves the HTML interface."""
-    return FileResponse("index.html")
+    return send_from_directory(os.getcwd(), 'index.html')
 
-@app.post("/upload_pdf/")
-async def upload_pdf(file_path: str):
+@app.route("/upload_pdf/", methods=["POST"])
+def upload_pdf():
     """Handles PDF file processing."""
     global documents
+    file = request.files['file']
+
+    # Save the uploaded file temporarily
+    file_path = os.path.join(os.getcwd(), file.filename)
+    file.save(file_path)
 
     # Extract text from PDF
     doc = fitz.open(file_path)
@@ -46,13 +49,15 @@ async def upload_pdf(file_path: str):
     embeddings = embed_model.encode(text_chunks)
     index.add(np.array(embeddings, dtype=np.float32))
 
-    return JSONResponse({"message": "PDF uploaded and indexed successfully!"})
+    return jsonify({"message": "PDF uploaded and indexed successfully!"})
 
-@app.get("/chat/")
-def chat_with_pdf(msg: str = Query(..., title="User Message")):
+@app.route("/chat/", methods=["GET"])
+def chat_with_pdf():
     """Handles user queries and returns AI-generated responses."""
+    msg = request.args.get("msg")
+
     if not documents:
-        return JSONResponse({"response": "Please upload a PDF first."})
+        return jsonify({"response": "Please upload a PDF first."})
 
     # Retrieve relevant context
     query_embedding = embed_model.encode([msg])
@@ -76,7 +81,7 @@ def chat_with_pdf(msg: str = Query(..., title="User Message")):
         token = chunk.choices[0].delta.content or ""
         response_text += token
 
-    return JSONResponse({"response": response_text})
+    return jsonify({"response": response_text})
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=8000)
+    app.run(host="0.0.0.0", port=8000)