few shorts upgrade

z_generate.py

@@ -87,10 +87,33 @@ Respond only to the question asked. Provide only number(s) of the source images
 
 If the image is relevant to the question then output format should be a list [1, 3, 0] 
 
-otherwise reply with []  (empty list)""",
+otherwise just reply empty list that is []""",
 
             },
 
+            {
+                "role": "user",
+                "content": """Context:
+Extracted Images:
+
+Document 0:::
+Rahuls playing football
+
+Document 1:::
+Rahul recieving award in Archery.
+
+---
+
+Now here is the question you need to answer.
+
+Question: Who is Rahul"""
+
+            },
+            {
+                "role": "assistant", 
+                "content": "[1, ]"
+            },
+
             {
                 "role": "user",
                 "content": """Context: