Update document_generator_v2.py

document_generator_v2.py

@@ -607,26 +607,37 @@ async def generate_document_outline_endpoint(
     ai_client = AIClient()
     document_generator = DocumentGenerator(ai_client)
     vision_tools = VisionTools(ai_client)
-    
+
     try:
+        # Handle image processing
         image_context = ""
         if images:
             image_context = await vision_tools.extract_images_info(images)
-            
             # Store the image_context in the cache
-            cache_key = f"image_context_{conversation_id}"
-            await FastAPICache.get_backend().set(cache_key, image_context, expire=3600)  # Cache for 1 hour
-        
+            image_cache_key = f"image_context_{conversation_id}"
+            await FastAPICache.get_backend().set(image_cache_key, image_context, expire=3600)  # Cache for 1 hour
+
+        # Handle document processing using the new load_documents function
+        documents_list = []
+        if documents:
+            documents_list = await load_documents(documents)
+            # Store the documents_list in the cache
+            docs_cache_key = f"documents_list_{conversation_id}"
+            await FastAPICache.get_backend().set(docs_cache_key, documents_list, expire=3600)  # Cache for 1 hour
+
+        # Generate document outline
         json_document = document_generator.generate_document_outline(
             query,
             template,
-            image_context=image_context
+            image_context=image_context,
+            documents_context=documents_list
         )
-        
+
         if json_document is None:
             raise HTTPException(status_code=500, detail="Failed to generate a valid document outline")
-        
+
         return JsonDocumentResponse(json_document=json_document)
+
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))