add latex display

app.py

@@ -53,16 +53,18 @@ def run_GOT(image, got_mode, fine_grained_mode="", ocr_color="", ocr_box=""):
         elif got_mode == "format fine-grained OCR":
             res = model.chat(tokenizer, image_path, ocr_type='format', ocr_box=ocr_box, ocr_color=ocr_color, render=True, save_render_file=result_path)
 
-        if os.path.exists(result_path):
+        res_markdown = f"$$ {res} $$"
+
+        if "format" in got_mode and os.path.exists(result_path):
             with open(result_path, 'r') as f:
                 html_content = f.read()
             encoded_html = base64.b64encode(html_content.encode('utf-8')).decode('utf-8')
             iframe_src = f"data:text/html;base64,{encoded_html}"
             iframe = f'<iframe src="{iframe_src}" width="100%" height="600px"></iframe>'
             download_link = f'<a href="data:text/html;base64,{encoded_html}" download="result_{unique_id}.html">Download Full Result</a>'
-            return res, f"{download_link}<br>{iframe}"
+            return res_markdown, f"{download_link}<br>{iframe}"
         else:
-            return res, None
+            return res_markdown, None
     except Exception as e:
         return f"Error: {str(e)}", None
     finally:
@@ -154,7 +156,7 @@ with gr.Blocks() as demo:
             submit_button = gr.Button("Submit")
         
         with gr.Column():
-            ocr_result = gr.Textbox(label="GOT output")
+            ocr_result = gr.Markdown(label="GOT output")
 
     with gr.Column():
         html_result = gr.HTML(label="rendered html", show_label=True)