Update app.py

app.py

@@ -86,14 +86,29 @@ Lab Values:
     # Call LLM
     result = pipe(
         prompt,
-        max_new_tokens=1500,
+        max_new_tokens=2000,
         do_sample=True,
         temperature=0.3,
         top_p=0.9,
         return_full_text=False
     )
 
-    return result[0]["generated_text"].strip()
+    text = result[0]["generated_text"].strip()
+
+    # Split into left (sections 1–3) and right (sections 5–6)
+    left_sections = []
+    right_sections = []
+    capture_left = True
+
+    for line in text.splitlines():
+        if line.strip().startswith("5. Tabular Mapping"):
+            capture_left = False
+        if capture_left:
+            left_sections.append(line)
+        else:
+            right_sections.append(line)
+
+    return "\n".join(left_sections), "\n".join(right_sections)
 
 
 # -----------------------
@@ -130,14 +145,19 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             
             analyze_btn = gr.Button("🔬 Generate Report", variant="primary")
     
-    gr.Markdown("### 📊 Analysis Output")
-    output = gr.Markdown(label="Medical Report")
+    with gr.Row():
+        with gr.Column(scale=1):
+            gr.Markdown("### 📝 Summary & Action Plan")
+            left_output = gr.Markdown()
+        with gr.Column(scale=1):
+            gr.Markdown("### 📊 Tabular & AI Insights")
+            right_output = gr.Markdown()
     
     analyze_btn.click(
         fn=analyze,
         inputs=[albumin, creatinine, glucose, crp, mcv, rdw, alp,
                 wbc, lymph, age, gender, height, weight],
-        outputs=output
+        outputs=[left_output, right_output]
     )
     
     gr.Markdown(