Update app.py

app.py

@@ -495,27 +495,27 @@ with gr.Blocks(css=css) as demo:
                         [txt_message, chat_bot, app_session]
                     )
 
-                # with gr.Tab("Few Shot") as fewshot_tab:
-                #     fewshot_tab_label = gr.Textbox(value="Few Shot", interactive=False, visible=False)
-                #     with gr.Row():
-                #         with gr.Column(scale=1):
-                #             image_input = gr.Image(type="filepath", sources=["upload"])
-                #         with gr.Column(scale=3):
-                #             user_message = gr.Textbox(label="User")
-                #             assistant_message = gr.Textbox(label="Assistant")
-                #             with gr.Row():
-                #                 add_demonstration_button = gr.Button("Add Example")
-                #                 generate_button = gr.Button(value="Generate", variant="primary")
-                #     add_demonstration_button.click(
-                #         fewshot_add_demonstration,
-                #         [image_input, user_message, assistant_message, chat_bot, app_session],
-                #         [image_input, user_message, assistant_message, chat_bot, app_session]
-                #     )
-                #     generate_button.click(
-                #         fewshot_respond,
-                #         [image_input, user_message, chat_bot, app_session, params_form],
-                #         [image_input, user_message, assistant_message, chat_bot, app_session]
-                #     )
+                with gr.Tab("Few Shot", visible=False) as fewshot_tab:
+                    fewshot_tab_label = gr.Textbox(value="Few Shot", interactive=False, visible=False)
+                    with gr.Row():
+                        with gr.Column(scale=1):
+                            image_input = gr.Image(type="filepath", sources=["upload"])
+                        with gr.Column(scale=3):
+                            user_message = gr.Textbox(label="User")
+                            assistant_message = gr.Textbox(label="Assistant")
+                            with gr.Row():
+                                add_demonstration_button = gr.Button("Add Example")
+                                generate_button = gr.Button(value="Generate", variant="primary")
+                    add_demonstration_button.click(
+                        fewshot_add_demonstration,
+                        [image_input, user_message, assistant_message, chat_bot, app_session],
+                        [image_input, user_message, assistant_message, chat_bot, app_session]
+                    )
+                    generate_button.click(
+                        fewshot_respond,
+                        [image_input, user_message, chat_bot, app_session, params_form],
+                        [image_input, user_message, assistant_message, chat_bot, app_session]
+                    )
 
                 chat_tab.select(
                     select_chat_type,
@@ -527,16 +527,16 @@ with gr.Blocks(css=css) as demo:
                     [txt_message, chat_bot, app_session],
                     [txt_message, chat_bot, app_session, image_input, user_message, assistant_message]
                 )
-                # fewshot_tab.select(
-                #     select_chat_type,
-                #     [fewshot_tab_label, app_session],
-                #     [app_session]
-                # )
-                # fewshot_tab.select( # do clear
-                #     clear,
-                #     [txt_message, chat_bot, app_session],
-                #     [txt_message, chat_bot, app_session, image_input, user_message, assistant_message]
-                # )
+                fewshot_tab.select(
+                    select_chat_type,
+                    [fewshot_tab_label, app_session],
+                    [app_session]
+                )
+                fewshot_tab.select( # do clear
+                    clear,
+                    [txt_message, chat_bot, app_session],
+                    [txt_message, chat_bot, app_session, image_input, user_message, assistant_message]
+                )
                 chat_bot.flushed(
                     flushed,
                     outputs=[txt_message]