Update app.py

app.py

@@ -47,23 +47,23 @@ def llm_response(history,text,img):
 
 # Function that takes User Inputs and displays it on ChatUI
 text_box_01 = "what is in the image"
-def output_query_message(txt,img):
+def output_query_message(img):
     if not img:
-        return txt
+        return text_box_01
     base64 = image_to_base64(img)
     data_url = f"data:image/jpeg;base64,{base64}"
-    outputText = [(f"{txt} ![]({data_url})", None)]
+    outputText = [(f"{text_box_01} ![]({data_url})", None)]
     return outputText
 
 # Function that takes User Inputs, generates Response and displays on Chat UI
-def output_llm_response(text,img):
+def output_llm_response(img):
     if not img:
-        response = txt_model.generate_content(text)
+        response = txt_model.generate_content(text_box_01)
         return response.text
 
     else:
         img = PIL.Image.open(img)
-        response = vis_model.generate_content([text,img])
+        response = vis_model.generate_content([text_box_01,img])
         return response.text
 
         
@@ -82,10 +82,10 @@ with gr.Blocks(theme='snehilsanyal/scikit-learn') as app1:
         
     btn = gr.Button("Check This")
     clicked = btn.click(output_query_message,
-                        [text_box_01,image_box],
+                        [image_box],
                         outputbox
                         ).then(output_llm_response,
-                                [text_box_01,image_box],
+                                [image_box],
                                 outputbox
                                 )
     gr.Markdown("""