Update app.py

app.py

@@ -28,7 +28,7 @@ INPUT: This file is a PDF that first contains the Question Paper and immediately
 TASK:  
 1. Transcribe EXACTLY all the questions FIRST (with their total marks).   
 2. After ALL questions, transcribe the Markscheme exactly, preserving M/A/R notation in brackets.   
-3. Always number the questions sequentially (Question 1, Question 2, Question 3a,Question 3b …) **in the order they appear in the PDF**, even if the PDF shows a different number or leaves it blank. Do NOT skip or leave Question: blank. Never start a question other than question 1 ( even if it is labelled in pdf as 8 name it 1) 
+3. Always number the questions sequentially (Question 1, Question 2, Question 3, …) **in the order they appear in the PDF**, even if the PDF shows a different number or leaves it blank. Do NOT skip or leave Question: blank. Never start a question other than question 1 ( even if it is labelled in pdf as 8 name it 1) 
 4. After the markscheme, DETECT and FLAG all questions in the markscheme where a graph/diagram is expected. For each, output the question number and the page number in the format below. 
 FORMAT:   
 ==== PAPER TOTAL MARKS ====   
@@ -177,14 +177,31 @@ def merge_pdfs(paths, output_path):
 
 def gemini_generate_content(model, prompt_text, file_upload_obj=None, image_obj=None): 
     """ 
-    Send prompt_text and optionally an uploaded file (or an image object) to the model. 
+    Send prompt_text and optionally an uploaded file (or an image object/list) to the model. 
     Returns textual response and prints progress. 
     """ 
     inputs = [prompt_text] 
     if file_upload_obj: 
         inputs.append(file_upload_obj) 
     if image_obj: 
-        inputs.append(image_obj) 
+        # Handle both single images and lists of images
+        if isinstance(image_obj, list):
+            # Convert image paths to PIL Image objects
+            for img_path in image_obj:
+                if isinstance(img_path, str):
+                    # It's a file path, load as PIL Image
+                    pil_img = Image.open(img_path)
+                    inputs.append(pil_img)
+                else:
+                    # It's already an image object
+                    inputs.append(img_path)
+        else:
+            # Single image
+            if isinstance(image_obj, str):
+                pil_img = Image.open(image_obj)
+                inputs.append(pil_img)
+            else:
+                inputs.append(image_obj)
     print("📡 Sending request to Gemini (prompt length:", len(prompt_text), "chars )") 
     response = model.generate_content(inputs) 
     raw_text = getattr(response, "text", None)