Update app.py

app.py

@@ -4,6 +4,7 @@ from typing import List
 from pathlib import Path
 import shutil
 import tempfile
+import os
 
 from langchain_docling import DoclingLoader
 from langchain_docling.loader import ExportType
@@ -24,16 +25,24 @@ async def upload_file(file: UploadFile = File(...)):
     # with file_path.open("wb") as buffer:
     #     shutil.copyfileobj(file.file, buffer)
 
-    with tempfile.NamedTemporaryFile(delete=False, suffix=file.filename) as temp_file:
-        # Efficiently write the uploaded file's content to the temporary file
-        contents = await file.read()
-        temp_file.write(contents)
+    # with tempfile.NamedTemporaryFile(delete=False, suffix=file.filename) as temp_file:
+    #     # Efficiently write the uploaded file's content to the temporary file
+    #     contents = await file.read()
+    #     temp_file.write(contents)
             
-    temp_file_path = temp_file.name 
+    # temp_file_path = temp_file.name 
 
-    # result = process_with_langchain(file_path)
+    suffix = os.path.splitext(file.filename)[-1] or ".pdf"
+    with tempfile.NamedTemporaryFile(delete=False, suffix=suffix, dir="/tmp") as tmp:
+        shutil.copyfileobj(file.file, tmp)
+        tmp_path = tmp.name
+
+    # At this point, tmp_path is a real file path in /tmp
+    # Debug: check if file is valid
+    size = os.path.getsize(tmp_path)
+    print(f"Saved {file.filename} -> {tmp_path} ({size} bytes)")
     
-    loader = DoclingLoader(file_path=str(temp_file_path), export_type=ExportType.MARKDOWN)
+    loader = DoclingLoader(file_path=str(tmp_path), export_type=ExportType.MARKDOWN)
     docs = loader.load()
     # docs = docs.model_dump()
     return {