Update app_main.py

app_main.py

@@ -9,6 +9,8 @@ from dotenv import load_dotenv
 import pytesseract
 from transformers import AutoProcessor, AutoModelForImageTextToText
 from langchain_community.document_loaders.image_captions import ImageCaptionLoader
+from werkzeug.utils import secure_filename
+import tempfile
 
 app = Flask(__name__)
 
@@ -126,27 +128,33 @@ def index():
     return render_template('app_index.html')
 
 # API endpoint
-@app.route('/process_static_pdf', methods=['POST'])
-def process_static_pdf():
-    # Option 1: Use hardcoded static PDF
-    pdf_path = r"E:\Pratham\2025\Harsh Sir\Scratch Vision\images\scratch_crab.pdf"
-
-    # Optional: Allow override via JSON request body
-    if request.json and "pdf_path" in request.json:
-        pdf_path = request.json["pdf_path"]
-
-    if not os.path.isfile(pdf_path):
-        return jsonify({"error": f"File not found: {pdf_path}"}), 400
-
-    # json_path = os.path.join(JSON_FOLDER_PATH, "extracted.json")
-    json_path = None
-    output_path, result = extract_images_from_pdf(pdf_path, json_path)
-    
-    return jsonify({
-        "message": "✅ PDF processed successfully",
-        "output_json": output_path,
-        "sprites": result
-    })
+@app.route('/process_pdf', methods=['POST'])
+def process_pdf():
+    try:
+        if 'pdf_file' not in request.files:
+            return jsonify({"error": "Missing PDF file in form-data with key 'pdf_file'"}), 400
+
+        pdf_file = request.files['pdf_file']
+        if pdf_file.filename == '':
+            return jsonify({"error": "Empty filename"}), 400
+
+        # Save the uploaded PDF temporarily
+        filename = secure_filename(pdf_file.filename)
+        temp_dir = tempfile.mkdtemp()
+        saved_pdf_path = os.path.join(temp_dir, filename)
+        pdf_file.save(saved_pdf_path)
+
+        # Extract & process
+        json_path = None
+        output_path, result = extract_images_from_pdf(saved_pdf_path, json_path)
+
+        return jsonify({
+            "message": "✅ PDF processed successfully",
+            "output_json": output_path,
+            "sprites": result
+        })
+    except Exception as e:
+        return jsonify({"error": f"❌ Failed to process PDF: {str(e)}"}), 500
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)