Update app.py

app.py

@@ -1,5 +1,6 @@
 import os
 import gradio as gr
+import markdown
 from openai import OpenAI
 
 # --- Initialize Hugging Face router client ---
@@ -13,9 +14,9 @@ client = OpenAI(
     api_key=HF_TOKEN,
 )
 
-# --- Chat handler ---
+# --- AI processing function ---
 def generate_report(age, gender, height, weight, albumin, creatinine, glucose, crp, mcv, rdw, alp, wbc, lymphocytes, hb, pv):
-    # --- System Prompt ---
+    # --- System prompt ---
     system = """You are an advanced Medical Insight Generation AI trained to analyze clinical biomarkers, urine analysis, and lab test results.
 Your goal is to generate a medically accurate, empathetic, and client-friendly health report in the following structured format:
 
@@ -33,7 +34,7 @@ Your goal is to generate a medically accurate, empathetic, and client-friendly h
 Maintain a professional, compassionate tone and explain medical reasoning in accessible language.
 """
 
-    # --- User message ---
+    # --- Format user message ---
     user_message = (
         f"Patient Info:\n"
         f"- Age: {age}\n"
@@ -55,33 +56,40 @@ Maintain a professional, compassionate tone and explain medical reasoning in acc
     )
 
     try:
-        # --- Send request to model ---
+        # --- Query model ---
         response = client.chat.completions.create(
             model="openai/gpt-oss-120b:fireworks-ai",
             messages=[
                 {"role": "system", "content": system},
                 {"role": "user", "content": user_message},
             ],
-            temperature=0.7,
-            max_tokens=2000,
+            temperature=0.5,
         )
+
+        # --- Get model reply and convert Markdown → HTML ---
         reply = response.choices[0].message.content
+        html_output = markdown.markdown(
+            reply,
+            extensions=["tables", "fenced_code", "nl2br"]
+        )
+
     except Exception as e:
-        reply = f"⚠️ Error: {str(e)}"
+        html_output = f"<p style='color:red;'>⚠️ Error: {str(e)}</p>"
 
-    return reply
+    return html_output
 
 
-# --- Gradio UI ---
+# --- Gradio Interface ---
 with gr.Blocks(title="🧬 Biomarker Medical Insight Chatbot") as demo:
     gr.Markdown(
         """
         ## 🧠 AI-Powered Biomarker Report Generator  
         Enter the patient details and biomarkers below.  
-        The AI will generate a **comprehensive medical report** with insights, risk assessment, and personalized recommendations.
+        The AI will generate a **comprehensive medical report** with structured insights, risk assessment, and recommendations.
         """
     )
 
+    # --- Basic Info ---
     with gr.Row():
         age = gr.Number(label="Age", value=45)
         gender = gr.Radio(["Male", "Female"], label="Gender", value="Male")
@@ -90,8 +98,8 @@ with gr.Blocks(title="🧬 Biomarker Medical Insight Chatbot") as demo:
         height = gr.Number(label="Height (cm)", value=175)
         weight = gr.Number(label="Weight (kg)", value=72)
 
+    # --- Biomarkers ---
     gr.Markdown("### 🧫 Biomarker Inputs (Demo Values Pre-filled)")
-
     with gr.Row():
         albumin = gr.Number(label="Albumin (g/dL)", value=4.2)
         creatinine = gr.Number(label="Creatinine (mg/dL)", value=1.1)
@@ -111,8 +119,9 @@ with gr.Blocks(title="🧬 Biomarker Medical Insight Chatbot") as demo:
         hb = gr.Number(label="Hemoglobin (g/dL)", value=14.5)
         pv = gr.Number(label="Plasma (PV) (mL)", value=3000)
 
+    # --- Submit + Output ---
     submit_btn = gr.Button("📤 Generate Medical Report")
-    output_box = gr.Textbox(label="AI-Generated Medical Report", lines=25)
+    output_box = gr.HTML(label="🧠 AI-Generated Medical Report (Rendered in Markup)")
 
     submit_btn.click(
         generate_report,