Update app.py

app.py

@@ -27,15 +27,17 @@ def generate_testing_instructions(images, context):
 
     base64_images = encode_images(images)
 
-    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. Each test case should include the following details:
+    # 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
+    ]
+    
+    # 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:
 
 - **Description**: A brief explanation of the feature being tested.
 - **Pre-conditions**: The required setup or state of the app before beginning the test.
@@ -51,24 +53,28 @@ 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.
-
-'''},
-                {"type": "text", "text": context}
-            ]
-        }
-        for base64_image in base64_images
-    ],
-    temperature=0,
-    max_tokens=1024,
-    top_p=1,
-    stream=False,
-    stop=None,
-)
+'''}
+    
+    # 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
 
-# Create the Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("## App Testing Instructions Generator")
+    gr.Markdown("## 🚌 Red Bus App Testing Instructions Generator")
 
     with gr.Row():
         context = gr.Textbox(label="Optional Context", placeholder="Add any specific instructions or questions...")
@@ -85,5 +91,4 @@ with gr.Blocks() as demo:
         outputs=output
     )
 
-# Launch the Gradio app
 demo.launch(debug=True)