Better PDF upload paths

backend/main.py

@@ -117,39 +117,53 @@ def process_file(file_path: str, file_name: str):
 
 @app.post("/upload")
 async def upload_document(file: UploadFile = File(...)):
-    if not file.filename.endswith('.pdf'):
+    if not file.filename.lower().endswith('.pdf'):
         raise HTTPException(status_code=400, detail="Only PDF files are supported")
     
     try:
-        # Create a temporary file
-        with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as temp_file:
-            content = await file.read()
-            temp_file.write(content)
-            temp_path = temp_file.name
-
-        # Process the file
-        texts = process_file(temp_path, file.filename)
+        # Read the file content directly into memory
+        content = await file.read()
         
-        # Create a new session
-        session_id = str(uuid4())
-        vector_db = VectorDatabase()
-        await vector_db.abuild_from_list(texts)
+        # Create a temporary file in a directory we know exists
+        temp_dir = "/tmp"  # Using /tmp which is writable in most environments
+        os.makedirs(temp_dir, exist_ok=True)
         
-        # Store session data
-        sessions[session_id] = {
-            "vector_db": vector_db,
-            "texts": texts
-        }
+        temp_path = os.path.join(temp_dir, f"upload_{file.filename}")
         
-        # Clean up
-        os.unlink(temp_path)
+        # Write the content to the temporary file
+        with open(temp_path, 'wb') as temp_file:
+            temp_file.write(content)
         
-        return {
-            "session_id": session_id,
-            "message": f"Document processed successfully. Added {len(texts)} chunks to the database."
-        }
+        try:
+            # Process the file
+            texts = process_file(temp_path, file.filename)
+            
+            # Create a new session
+            session_id = str(uuid4())
+            vector_db = VectorDatabase()
+            await vector_db.abuild_from_list(texts)
+            
+            # Store session data
+            sessions[session_id] = {
+                "vector_db": vector_db,
+                "texts": texts
+            }
+            
+            return {
+                "session_id": session_id,
+                "message": f"Document processed successfully. Added {len(texts)} chunks to the database."
+            }
+            
+        finally:
+            # Clean up the temporary file
+            try:
+                if os.path.exists(temp_path):
+                    os.unlink(temp_path)
+            except Exception as e:
+                print(f"Warning: Could not delete temporary file: {e}")
+                
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=f"Error processing file: {str(e)}")
 
 @app.post("/query/{session_id}")
 async def query_documents(session_id: str, query: Query):