Update app.py

app.py

@@ -1,25 +1,40 @@
 import gradio as gr
 import numpy as np
+import matplotlib.pyplot as plt
 
-def calculate_tia(t1, a1, t2, a2):
+def calculate_and_plot(t1, a1, t2, a2):
     try:
-        # Calculate washout rate constant (k)
+        # الحسابات الأساسية
         k = np.log(a1 / a2) / (t2 - t1)
-        # Estimate initial activity (A₀)
         a0 = a1 / np.exp(-k * t1)
-        # Calculate Time-Integrated Activity (TIA)
         tia = a0 / k
 
         result = f"""
         ### ✅ Results
         - **A₀ (initial activity):** {a0:.2f} MBq  
         - **k (washout rate):** {k:.5f} / hour  
-        - **TIA (time-integrated activity):** {tia:.2f} MBq·h  
+        - **TIA:** {tia:.2f} MBq·h  
         """
-    except Exception:
-        result = "⚠️ Error: Please check your inputs. Values must be positive and non-zero."
 
-    return result
+        # إنشاء نقاط الرسم
+        t_points = np.linspace(0, max(t2 + 40, 200), 100)
+        a_points = a0 * np.exp(-k * t_points)
+
+        # رسم المنحنى باستخدام matplotlib
+        fig, ax = plt.subplots(figsize=(6, 4))
+        ax.plot(t_points, a_points, label="Fitted Curve", color='blue')
+        ax.scatter([t1, t2], [a1, a2], color='red', label="Input Points")
+        ax.set_xlabel("Time (hours)")
+        ax.set_ylabel("Activity (MBq)")
+        ax.set_title("Mono-Exponential Washout Curve")
+        ax.grid(True)
+        ax.legend()
+        plt.tight_layout()
+
+        return result, fig
+
+    except Exception:
+        return "⚠️ Error: Please check your inputs.", None
 
 with gr.Blocks() as demo:
     gr.Markdown("# ⚛️ Lu-177 Dosimetry Tool\nEstimate Time-Integrated Activity (TIA) from Two SPECT Time Points")
@@ -32,9 +47,11 @@ with gr.Blocks() as demo:
         t2 = gr.Number(label="Time Point 2 (hours)", value=60)
         a2 = gr.Number(label="Activity at Time 2 (MBq)", value=4)
 
-    output = gr.Markdown()
-    button = gr.Button("Calculate TIA")
-    button.click(fn=calculate_tia, inputs=[t1, a1, t2, a2], outputs=output)
+    output_text = gr.Markdown()
+    output_plot = gr.Plot()
+    button = gr.Button("Calculate & Plot TIA")
+
+    button.click(fn=calculate_and_plot, inputs=[t1, a1, t2, a2], outputs=[output_text, output_plot])
 
     gr.Markdown("""
     ---
@@ -66,5 +83,5 @@ with gr.Blocks() as demo:
 
     ---
     """)
-    
+
 demo.launch()