Update app.py

app.py

@@ -8,12 +8,10 @@ import secrets
 import dashscope
 from dashscope import MultiModalConversation, Generation
 
-# 设置API密钥
 YOUR_API_TOKEN = os.getenv('YOUR_API_TOKEN')
 dashscope.api_key = YOUR_API_TOKEN
 math_messages = []
 def process_image(image):
-    # 获取上传文件的目录
     global math_messages
     math_messages = [] # reset when upload image
     uploaded_file_dir = os.environ.get("GRADIO_TEMP_DIR") or str(
@@ -21,14 +19,12 @@ def process_image(image):
     )
     os.makedirs(uploaded_file_dir, exist_ok=True)
     
-    # 创建临时文件路径
     name = f"tmp{secrets.token_hex(20)}.jpg"
     filename = os.path.join(uploaded_file_dir, name)
-    # 保存上传的图片
     image.save(filename)
 
     
-    # 调用qwen-vl-max-0809模型处理图片
+    # Use qwen-vl-max-0809 for OCR
     messages = [{
         'role': 'system',
         'content': [{'text': 'You are a helpful assistant.'}]
@@ -42,7 +38,6 @@ def process_image(image):
     
     response = MultiModalConversation.call(model='qwen-vl-max-0809', messages=messages)
     
-    # 清理临时文件
     os.remove(filename)
     
     return response.output.choices[0]["message"]["content"]
@@ -90,7 +85,7 @@ css = """
 #qwen-md .katex-display>.katex>.katex-html { display: inline; }
 """
 
-# 创建Gradio接口
+# Crate interface
 iface = gr.Interface(
     css=css,
     fn=math_chat_bot,
@@ -107,10 +102,13 @@ iface = gr.Interface(
         {"left": "\\begin\{CD\}", "right": "\\end\{CD\}", "display": True},
         {"left": "\\[", "right": "\\]", "display": True}
     ], elem_id="qwen-md"),
-    title="Qwen2-math demo",
+    title=gr.Markdown("""<center><font size=8>📖 Qwen2 Math Demo</center>"""),
     allow_flagging='never',
-    description="upload a math question image and ask your question.(Supports only formulas and text, not geometry etc.)"
+    description=gr.Markdown(
+            """\
+<center><font size=3>This WebUI is based on Qwen2-VL for OCR and Qwen2-Math for mathematical reasoning. You can input either images or texts of mathematical or arithmetic problems.</center><br>\
+    """)
 )
 
-# 启动Gradio应用
+# Launch gradio application
 iface.launch()