Update app.py

app.py

@@ -76,19 +76,15 @@ def math_chatbot(image, sketchpad, question, state):
     return get_math_response(image_description, question)
 
 
+
+def tabs_select(e: gr.SelectData, _state):
+    _state["tab_index"] = e.index
+
 css = """
 #qwen-md .katex-display { display: inline; }
 #qwen-md .katex-display>.katex { display: inline; }
 #qwen-md .katex-display>.katex>.katex-html { display: inline; }
 """
-
-
-# Tab selection callback
-def tabs_select(e: gr.SelectData, _state):
-    _state["tab_index"] = e.index
-
-
-# Gradio interface
 with gr.Blocks(css=css) as demo:
     gr.HTML("""\
 <p align="center"><img src="https://huggingface.co/front/assets/huggingface_logo.svg" style="height: 60px"/><p>"""
@@ -97,24 +93,23 @@ with gr.Blocks(css=css) as demo:
 <center><font size=3>This demo uses Hugging Face models for OCR and mathematical reasoning. You can input images or text-based questions.</center>"""
             )
     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(label="Sketch", layers=False)
+                    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_text = gr.Textbox(label="input your question")
             with gr.Row():
                 with gr.Column():
-                    clear_btn = gr.ClearButton([input_image, input_sketchpad, input_text])
+                    clear_btn = gr.ClearButton(
+                        [*input_image, input_sketchpad, input_text])
                 with gr.Column():
                     submit_btn = gr.Button("Submit", variant="primary")
-
         with gr.Column():
-            output_md = gr.Markdown(label="Answer",
+            output_md = gr.Markdown(label="answer",
                                     latex_delimiters=[{
                                         "left": "\\(",
                                         "right": "\\)",
@@ -123,18 +118,30 @@ with gr.Blocks(css=css) as demo:
                                         "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_chatbot,
-        inputs=[input_image, input_sketchpad, input_text, state],
-        outputs=output_md
-    )
-
-# Launch Gradio app
-demo.launch()
+        submit_btn.click(
+            fn=math_chat_bot,
+            inputs=[*input_image, input_sketchpad, input_text, state],
+            outputs=output_md)
+demo.launch()