Update app.py

app.py

@@ -735,8 +735,8 @@ def upload_callback(image_input, state, visual_chatgpt=None, openai_api_key=None
     )
 ]
 
-    return state, state, image_input, click_state, image_input, image_input, image_input, image_embedding, \
-        original_size, input_size, f"Name: {name}", f"Artist: {artist}", f"Year: {year}", f"Material: {material}",f"Name: {name}", f"Artist: {artist}", f"Year: {year}", f"Material: {material}",paragraph,artist
+    return [state, state, image_input, click_state, image_input, image_input, image_input, image_input, image_embedding, \
+        original_size, input_size] + [f"Name: {name}", f"Artist: {artist}", f"Year: {year}", f"Material: {material}"]*4 + [paragraph,artist]
 
 
 
@@ -1154,6 +1154,8 @@ def cap_everything_withoutsound(image_input, visual_chatgpt, text_refiner,paragr
     # visual_chatgpt.agent.memory.buffer = visual_chatgpt.agent.memory.buffer + visual_chatgpt.global_prompt 
     return paragraph
 
+
+
 # def handle_liked(state,like_res):
 #     if state:
 #         like_res.append(state[-1][1])
@@ -1378,6 +1380,14 @@ def create_ui():
                             artist_label_base = gr.Button(value="Artist: ")
                             year_label_base = gr.Button(value="Year: ")
                             material_label_base = gr.Button(value="Material: ")   
+                    
+                    with gr.Tab("Base2") as base_tab2:
+                        image_input_base_2 = gr.Image(type="pil", interactive=True, elem_id="image_upload")
+                        with gr.Row():
+                            name_label_base2 = gr.Button(value="Name: ")
+                            artist_label_base2 = gr.Button(value="Artist: ")
+                            year_label_base2 = gr.Button(value="Year: ")
+                            material_label_base2 = gr.Button(value="Material: ")  
 
                     with gr.Tab("Click") as click_tab:
                         image_input = gr.Image(type="pil", interactive=True, elem_id="image_upload")
@@ -1423,6 +1433,11 @@ def create_ui():
                         #                                elem_id="image_sketcher")
                         sketcher_input = gr.ImageEditor(type="pil", interactive=True, 
                                                        elem_id="image_sketcher")
+                        with gr.Row():
+                            name_label_traj = gr.Button(value="Name: ")
+                            artist_label_traj = gr.Button(value="Artist: ")
+                            year_label_traj = gr.Button(value="Year: ")
+                            material_label_traj = gr.Button(value="Material: ")
                         # example_image_traj = gr.Image(type="pil", interactive=False, visible=False)
                         with gr.Row():                   
                             clear_button_sketcher = gr.Button(value="Clear Sketch", interactive=True)
@@ -1895,8 +1910,10 @@ def create_ui():
         
 
         image_input_base.upload(upload_callback, [image_input_base, state, visual_chatgpt,openai_api_key],
-                           [chatbot, state, origin_image, click_state, image_input, image_input_base, sketcher_input,
-                            image_embedding, original_size, input_size,name_label,artist_label,year_label,material_label,name_label_base, artist_label_base, year_label_base, material_label_base,paragraph,artist])
+                           [chatbot, state, origin_image, click_state, image_input, image_input_base, sketcher_input,image_input_base_2,
+                            image_embedding, original_size, input_size,name_label,artist_label,year_label,material_label,name_label_base, artist_label_base, year_label_base, material_label_base, \
+                            name_label_base2, artist_label_base2, year_label_base2, material_label_base2,name_label_traj, artist_label_traj, year_label_traj, material_label_traj, \
+                                paragraph,artist])
         
         # image_input.upload(upload_callback, [image_input, state, visual_chatgpt, openai_api_key],
         #                    [chatbot, state, origin_image, click_state, image_input, image_input_base, sketcher_input,
@@ -1911,8 +1928,10 @@ def create_ui():
                           [chatbot, state, aux_state,output_audio])
         submit_button_text.click(lambda: "", None, chat_input)
         example_image.change(upload_callback, [example_image, state, visual_chatgpt, openai_api_key],
-                             [chatbot, state, origin_image, click_state, image_input, image_input_base, sketcher_input,
-                              image_embedding, original_size, input_size,name_label,artist_label,year_label,material_label,name_label_base, artist_label_base, year_label_base, material_label_base,paragraph,artist])
+                             [chatbot, state, origin_image, click_state, image_input, image_input_base, sketcher_input,image_input_base_2,
+                              image_embedding, original_size, input_size,name_label,artist_label,year_label,material_label,name_label_base, artist_label_base, year_label_base, material_label_base, \
+                            name_label_base2, artist_label_base2, year_label_base2, material_label_base2,name_label_traj, artist_label_traj, year_label_traj, material_label_traj, \
+                            paragraph,artist])
 
         example_image.change(clear_chat_memory, inputs=[visual_chatgpt])
 
@@ -1939,6 +1958,7 @@ def create_ui():
         traj_tab.select(on_click_tab_selected, outputs=[modules_need_gpt1,modules_not_need_gpt2,modules_need_gpt0,modules_need_gpt2])
         click_tab.select(on_click_tab_selected, outputs=[modules_need_gpt1,modules_not_need_gpt2,modules_need_gpt0,modules_need_gpt2])
         base_tab.select(on_base_selected, outputs=[modules_need_gpt0,modules_need_gpt2,modules_not_need_gpt2,modules_need_gpt1])
+        base_tab2.select(on_base_selected, outputs=[modules_not_need_gpt2,modules_not_need_gpt2,modules_need_gpt0,modules_need_gpt1])