Update app.py

app.py

@@ -103,61 +103,37 @@ css = """
 #qwen-md .katex-display>.katex { display: inline; }
 #qwen-md .katex-display>.katex>.katex-html { display: inline; }
 """
+
 def tabs_select(e: gr.SelectData, _state):
     _state["tab_index"] = e.index
+    return _state
 
 with gr.Blocks(css=css) as demo:
-
     state = gr.State({"tab_index": 0})
+
     with gr.Row():
         with gr.Column():
             with gr.Tabs() as input_tabs:
                 with gr.Tab("Upload"):
-                    input_image = gr.Image(type="pil", label="Upload"),
+                    input_image = gr.Image(type="pil", label="Upload")
                 with gr.Tab("Sketch"):
                     input_sketchpad = gr.Sketchpad(type="pil", label="Sketch", layers=False)
-            input_tabs.select(fn=tabs_select, inputs=[state])
-            input_text = gr.Textbox(label="input your question")
+            
+            input_tabs.select(fn=tabs_select, inputs=[state], outputs=[state])
+
+            input_text = gr.Textbox(label="Input your question")
             with gr.Row():
-                with gr.Column():
-                    clear_btn = gr.ClearButton(
-                        [*input_image, input_sketchpad, input_text])
-                with gr.Column():
-                    submit_btn = gr.Button("Submit", variant="primary")
+                clear_btn = gr.ClearButton([input_image, input_sketchpad, input_text])
+                submit_btn = gr.Button("Submit", variant="primary")
+
         with gr.Column():
-            output_md = gr.Markdown(label="answer",
-                                    latex_delimiters=[{
-                                        "left": "\\(",
-                                        "right": "\\)",
-                                        "display": True
-                                    }, {
-                                        "left": "\\begin\{equation\}",
-                                        "right": "\\end\{equation\}",
-                                        "display": True
-                                    }, {
-                                        "left": "\\begin\{align\}",
-                                        "right": "\\end\{align\}",
-                                        "display": True
-                                    }, {
-                                        "left": "\\begin\{alignat\}",
-                                        "right": "\\end\{alignat\}",
-                                        "display": True
-                                    }, {
-                                        "left": "\\begin\{gather\}",
-                                        "right": "\\end\{gather\}",
-                                        "display": True
-                                    }, {
-                                        "left": "\\begin\{CD\}",
-                                        "right": "\\end\{CD\}",
-                                        "display": True
-                                    }, {
-                                        "left": "\\[",
-                                        "right": "\\]",
-                                        "display": True
-                                    }],
-                                    elem_id="qwen-md")
-        submit_btn.click(
-            fn=math_chat_bot,
-            inputs=[*input_image, input_sketchpad, input_text, state],
-            outputs=output_md)
+            output_md = gr.Markdown(label="Answer", elem_id="qwen-md")
+
+    submit_btn.click(
+        fn=math_chat_bot,
+        inputs=[input_image, input_sketchpad, input_text, state],
+        outputs=output_md,
+        streaming=True
+    )
+
 demo.launch()