added multiple images support

app.py

@@ -32,21 +32,24 @@ def main():
 
         goal = st.text_area("Goal", placeholder="Enter the goal here")
 
-        image = st.file_uploader("Upload Image", type=['jpg', 'png'])
+        images = st.file_uploader("Upload Image", type=['jpg', 'png'], accept_multiple_files=True)
 
-        if image:
-            cols = st.columns(3)
-            with cols[1]:
-                st.image(image, caption="Uploaded Image", use_column_width=True)
+        if images:
+            cols = st.columns(len(images))
+
+            for i, image in enumerate(images):
+                with cols[i]:
+                    image = Image.open(image)
+                    st.image(image, caption="Uploaded Image", use_column_width=True)
         
 
         if st.button("Submit"):
-            if not story or not goal or (entity_opt and not entity) or not image:
+            if not story or not goal or (entity_opt and not entity) or not images:
                 st.error("Please fill all the fields")
                 return
             with col2:
                 with st.status("Generating response...", expanded=True):
-                    response = get_gpt4V_response(story, goal, entity, image, temperature=temperature)
+                    response = get_gpt4V_response(story, goal, entity, images, temperature=temperature)
                 
                 try:
                     response_json = get_str_to_json(response)

utils.py

@@ -68,10 +68,13 @@ data = {
 }
 
 
-def get_gpt4V_response(story, goal, entity, image, temperature=0.5):
+def get_gpt4V_response(story, goal, entity, images, temperature=0.5):
     # Convert image to base64
-    image_b64 = im_2_b64(image)
-    image_url = f"data:image/jpeg;base64,{image_b64.decode('utf-8')}"
+    image_urls = []
+    for i, image in enumerate(images):
+        image_b64 = im_2_b64(image)
+        image_url = f"data:image/jpeg;base64,{image_b64.decode('utf-8')}"
+        image_urls.append(image_url)
     st.write("✅ Image converted")
 
     if entity:
@@ -87,12 +90,13 @@ def get_gpt4V_response(story, goal, entity, image, temperature=0.5):
 
     st.write("✅ Prompt created")
 
-    content.append({
-        "type": "image_url",
-        "image_url": {
-            "url": image_url,
-        },
-    })
+    for image_url in image_urls:
+        content.append({
+            "type": "image_url",
+            "image_url": {
+                "url": image_url,
+            },
+        })
     
     st.write("🚀 Getting Response from GPT4V")
     response = client.chat.completions.create(