Update app.py

app.py

@@ -22,13 +22,32 @@ moondream.eval()
 def answer_questions(image_tuples, prompt_text):
     result = ""
     
-    prompts = [p.strip() for p in prompt_text.split(',')]  # Splitting and cleaning prompts    
+    prompts = [p.strip() for p in prompt_text.split(',')]  # Splitting and cleaning prompts
+    print(f"prompts\n{prompts}\n")
     image_embeds = [img[0] for img in image_tuples if img[0] is not None]  # Extracting images from tuples, ignoring None
     
     # Check if the lengths of image_embeds and prompts are equal
-    if len(image_embeds) != len(prompts):
-        return ("Error: The number of images input and prompts input (seperate by commas in input text field) must be the same.")
-        
+    #if len(image_embeds) != len(prompts):
+        #return ("Error: The number of images input and prompts input (seperate by commas in input text field) must be the same.")
+
+    answers = []
+    for prompt in prompts:
+        image_answers = moondream.batch_answer(
+            images=[img.convert("RGB") for img in image_embeds],
+            prompts=[prompt] * len(image_embeds),
+            tokenizer=tokenizer,
+        )
+        answers.append(image_answers)
+
+    for i in range(len(image_tuples)):
+        image_name = f"image{i+1}"
+        image_answers = [answer[i] for answer in answers]
+        print(f"image{i+1}_answers \n {image_answers} \n")
+        data.append([image_name] + image_answers)
+
+    result = {'headers': prompts, 'data': data}
+    return result
+    '''
     answers = moondream.batch_answer(
         images=image_embeds,
         prompts=prompts,
@@ -39,6 +58,7 @@ def answer_questions(image_tuples, prompt_text):
         result += (f"Q: {question}\nA: {answer}\n\n")
         
     return result
+    '''
 
 with gr.Blocks() as demo:
     gr.Markdown("# moondream2 unofficial batch processing demo")