Update app.py

app.py

@@ -59,7 +59,7 @@ async def generate_solution_python(chat_history):
                         latest_user_input = "an image"
                         break
                     # If a document was processed and its text added, use that
-                    if part.get("text") and part["text"].startswith("Document content:"):
+                    if part.get("text") and part["text"].startswith("PDF Document Content:") or part["text"].startswith("Document content:"):
                         latest_user_input = "a document"
                         break
             if latest_user_input:
@@ -122,6 +122,7 @@ def index():
 @app.route('/generate', methods=['POST'])
 async def generate():
     """Handles the AI generation request, managing conversation history and multi-modal input."""
+    session_id = None # Initialize session_id to None
     try:
         data = request.get_json()
         if not data:
@@ -131,13 +132,16 @@ async def generate():
         image_data = data.get('image_data') # Base64 image data
         document_text = data.get('document_text') # Text extracted from .txt on frontend
         pdf_data = data.get('pdf_data') # Base64 PDF data
-
-        if not (user_query or image_data or document_text or pdf_data):
-            return jsonify({"error": "Query, image, or document is required in the request body"}), 400
+        
+        # Ensure session_id is assigned before use
+        session_id = data.get('session_id')
         if not session_id:
             session_id = str(uuid.uuid4())
             print(f"Warning: session_id not provided, generated new one: {session_id}")
 
+        if not (user_query or image_data or document_text or pdf_data):
+            return jsonify({"error": "Query, image, or document is required in the request body"}), 400
+        
         current_chat_history = conversation_histories.get(session_id, [])
         
         # Construct the parts for the user message
@@ -178,7 +182,9 @@ async def generate():
             except Exception as pdf_error:
                 print(f"Error processing PDF: {pdf_error}")
                 user_message_parts.append({"text": f"PDF Document: (Error processing PDF: {pdf_error})"})
-                return jsonify({"error": f"Failed to process PDF: {pdf_error}"}), 400 # Return error to frontend
+                # Do not return error to frontend immediately for PDF processing issues
+                # Let the LLM try to respond even if PDF extraction failed
+                # return jsonify({"error": f"Failed to process PDF: {pdf_error}"}), 400 
 
         # If only a file was provided without a query, add a default instruction
         if not user_query and (image_data or document_text or pdf_data):
@@ -203,7 +209,11 @@ async def generate():
 
     except Exception as e:
         print(f"Error in /generate endpoint: {e}")
-        return jsonify({"error": f"Internal server error: {e}"}), 500
+        # Ensure session_id is handled even in the outer exception for logging/debugging
+        if session_id:
+            return jsonify({"error": f"Internal server error for session {session_id}: {e}"}), 500
+        else:
+            return jsonify({"error": f"Internal server error: {e}"}), 500
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)