Update app.py

app.py

@@ -6,38 +6,33 @@ import os
 
 client = Groq(api_key=os.environ.get('GROQ_API_KEY'))
 
-def encode_images(image_paths):
-    base64_images = []
-    for image_path in image_paths:
-        with open(image_path, "rb") as image_file:
-            base64_image = base64.b64encode(image_file.read()).decode("utf-8")
-            base64_images.append(f"data:image/jpeg;base64,{base64_image}")
-    return base64_images
-
-def resize_images(image_paths, target_size=(224, 224)):
-    resized_images = []
-    for image_path in image_paths:
-        img = Image.open(image_path)
-        img_resized = img.resize(target_size) 
-        resized_images.append(img_resized)
-    return resized_images
+def encode_image(image_path):
+    with open(image_path, "rb") as image_file:
+        base64_image = base64.b64encode(image_file.read()).decode("utf-8")
+    return f"data:image/jpeg;base64,{base64_image}"
 
-def generate_testing_instructions(images, context):
-    resized_images = resize_images(images)
-
-    base64_images = encode_images(images)
+def resize_image(image_path, target_size=(224, 224)):
+    img = Image.open(image_path)
+    img_resized = img.resize(target_size) 
+    return img_resized
 
-    # Combine all the base64 images into a single message
-    image_blocks = [
-        {"type": "image_url", "image_url": {"url": base64_image}}
-        for base64_image in base64_images
-    ]
+def generate_testing_instructions(images, context):
+    all_instructions = []
     
-    # Add the instructional text in the same message
-    content_block = {
-        "type": "text",
-        "text": '''
-You're a helpful assistant that creates comprehensive testing instructions. Based on the screenshot of app interface you should tell:
+    for image in images:
+        resized_image = resize_image(image)
+        base64_image = encode_image(image)
+        
+        # Send only one image and the text block in a single message
+        completion = client.chat.completions.create(
+            model="llava-v1.5-7b-4096-preview",
+            messages=[
+                {
+                    "role": "user",
+                    "content": [
+                        {"type": "image_url", "image_url": {"url": base64_image}},
+                        {"type": "text", "text": '''
+You're a helpful assistant that creates comprehensive testing instructions. Based on the screenshot of the app interface, you should tell:
 
 - **Description**: A brief explanation of the feature being tested.
 - **Pre-conditions**: The required setup or state of the app before beginning the test.
@@ -53,25 +48,21 @@ Please demonstrate your approach using the following features of a mobile app:
 5. **Offers**: Showcases available promotions and discounts.
 6. **Filters**: Options to filter buses by time, price, or other preferences.
 7. **Bus Information**: Provides details about the selected bus, including amenities, photos, and reviews.
-'''}
-    
-    # Single message combining all image URLs and text
-    completion = client.chat.completions.create(
-        model="llava-v1.5-7b-4096-preview",
-        messages=[
-            {
-                "role": "user",
-                "content": image_blocks + [content_block]  # Add images and text as one user message
-            }
-        ],
-        temperature=0,
-        max_tokens=1024,
-        top_p=1,
-        stream=False,
-        stop=None,
-    )
-    
-    return completion.choices[0].message.content
+
+''' + context}
+                    ]
+                }
+            ],
+            temperature=0,
+            max_tokens=1024,
+            top_p=1,
+            stream=False,
+            stop=None,
+        )
+        
+        all_instructions.append(completion.choices[0].message.content)
+
+    return "\n\n".join(all_instructions)
 
 with gr.Blocks() as demo:
     gr.Markdown("## 🚌 Red Bus App Testing Instructions Generator")