Working

api.py

@@ -97,8 +97,21 @@ async def upload(files: List[UploadFile] = File(...)):
         
         # Explicitly clear memory before processing new files
         logger.info("Clearing previous vector store from memory...")
+        old_store_had_value = store.get("value") is not None
         store["value"] = None
+        
+        # Force garbage collection
         gc.collect()
+        
+        # More aggressive memory cleanup if needed
+        if old_store_had_value:
+            try:
+                if hasattr(gc, 'collect'):
+                    for i in range(3):  # Run multiple collection cycles
+                        gc.collect(i)
+            except Exception as e:
+                logger.warning(f"Error during aggressive garbage collection: {e}")
+                
         logger.info("Memory cleared.")
 
         logger.info("Starting document processing...")
@@ -240,3 +253,30 @@ async def status():
     logger.info(f"Status response: {status_info}")
     
     return status_info
+
+@app.post("/clear")
+async def clear_context():
+    """Clear the current document context and free up memory."""
+    global store
+    logger.info("Clearing document context...")
+    
+    # Clear the store
+    if store.get("value") is not None:
+        store["value"] = None
+        
+        # Force garbage collection
+        gc.collect()
+        
+        # Run a more aggressive memory cleanup
+        try:
+            if hasattr(gc, 'collect'):
+                for i in range(3):  # Run multiple collection cycles
+                    gc.collect(i)
+        except Exception as e:
+            logger.warning(f"Error during aggressive garbage collection: {e}")
+        
+        logger.info("Document context cleared successfully.")
+        return {"status": "success", "message": "Document context cleared successfully."}
+    else:
+        logger.info("No document context to clear.")
+        return {"status": "info", "message": "No document context was loaded."}