Update app.py

app.py

@@ -10,6 +10,9 @@ import pathlib
 txt_model = genai.GenerativeModel('gemini-pro')
 vis_model = genai.GenerativeModel('gemini-pro-vision')
 
+txt_prompt_1 = 'The image contains two sample images, A and B. The first image (A) contains a lens, an anti-static strap, an Adrucam board, and a Raspberry Pi board. Which is missing on Image B?' 
+txt_display_1 = 'What are missing on B?'
+
 import os
 
 GOOGLE_API_KEY=os.getenv('GOOGLE_API_KEY')
@@ -46,19 +49,19 @@ def llm_response(history,text,img):
         return history
 
 # Function that takes User Inputs and displays it on ChatUI
-text_box_01 = "what is in the image"
+
 def output_query_message(img):
     if not img:
-        return text_box_01
+        return txt_prompt_1
     base64 = image_to_base64(img)
     data_url = f"data:image/jpeg;base64,{base64}"
-    outputText = [(f"{text_box_01} ![]({data_url})", None)]
+    outputText = [(f"{txt_display_1} ![]({data_url})", None)]
     return outputText
 
 # Function that takes User Inputs, generates Response and displays on Chat UI
 def output_llm_response(img):
     if not img:
-        response = txt_model.generate_content(text_box_01)
+        response = txt_model.generate_content(txt_prompt_1)
         return response.text
 
     else: