Update app.py

app.py

@@ -82,18 +82,31 @@ def index():
     """Serve the HTML page for the user interface"""
     return render_template('index.html')
 
+UPLOAD_FOLDER = "uploaded_files"
+os.makedirs(UPLOAD_FOLDER, exist_ok=True)  # Ensure the folder exists
+
 @app.route('/upload_pdf', methods=['POST'])
 def upload_pdf():
     """Handle PDF upload"""
+    if 'pdf' not in request.files:
+        return jsonify({"error": "No file part"}), 400  # Handle missing file
+
     file = request.files['pdf']
-    pdf_path = f"uploaded_files/{file.filename}"
-    file.save(pdf_path)
+    if file.filename == "":
+        return jsonify({"error": "No selected file"}), 400  # Handle empty filename
+
+    pdf_path = os.path.join(UPLOAD_FOLDER, file.filename)
+
+    try:
+        file.save(pdf_path)  # Save the uploaded PDF
 
-    # Extract text and create vector database
-    text_chunks = extract_text_from_pdf(pdf_path)
-    create_vector_db(text_chunks)
+        # Extract text and create vector database
+        text_chunks = extract_text_from_pdf(pdf_path)
+        create_vector_db(text_chunks)
 
-    return jsonify({"message": "PDF uploaded and indexed successfully!"})
+        return jsonify({"message": "PDF uploaded and indexed successfully!"}), 200
+    except Exception as e:
+        return jsonify({"error": f"Error processing file: {str(e)}"}), 500
 
 @app.route('/ask_question', methods=['POST'])
 def ask_question():