Update app.py

app.py

@@ -19,7 +19,16 @@ def generate_response(prompt_template, **kwargs):
     )
     response_content = response.choices[0].message["content"]
     print(response_content)
-    return gr.update(value=f"{response_content}")
+    html = markdown.markdown(
+        response_content,
+        extensions=[
+            "markdown.extensions.fenced_code",
+            "markdown.extensions.tables",
+            "markdown.extensions.attr_list",
+        ]
+    )
+    return f"""<script type="text/javascript" async src="https://cdnjs.cloudflare.com/ajax/libs/mathjax/2.7.7/MathJax.js?config=TeX-MML-AM_CHTML"></script>
+    {html}"""
 
 # Gradio app interface
 with gr.Blocks() as app:
@@ -44,7 +53,7 @@ with gr.Blocks() as app:
                     )
             # Button and output
             button = gr.Button(f"{tab_name} Execute")
-            output = gr.Markdown(label="Output", elem_id="latex-output")
+            output = gr.HTML()
             # Link button to the response wrapper
             button.click(
                 fn=lambda *args: generate_response(prompt_template, **dict(zip([inp["key"] for inp in inputs], args))),
@@ -99,14 +108,6 @@ with gr.Blocks() as app:
         ]
     )
 
-# Add custom CSS for LaTeX rendering
-app.css = """
-#latex-output {
-    font-family: "Computer Modern", serif;
-    font-size: 16px;
-    line-height: 1.5;
-}
-"""
 
 # Launch the app
 app.launch(debug=True)