Update app.py

app.py

@@ -13,7 +13,7 @@ def drake_simulator(R_star, f_p, n_e, f_l, f_i, f_c, L):
     intelligent = int(habitable * f_i)
     communicative = int(intelligent * f_c)
 
-    # Create a galaxy visualization
+    # Create galaxy plot
     fig, ax = plt.subplots(figsize=(6, 6))
     ax.set_facecolor("black")
     ax.set_title(f"Estimated Civilizations: {int(N)}", color="white")
@@ -44,9 +44,20 @@ def drake_simulator(R_star, f_p, n_e, f_l, f_i, f_c, L):
     com_y = int_y[:communicative]
     ax.scatter(com_x, com_y, color="red", s=8)
 
-    return fig
+    # Text description
+    annotation = f"""
+    🧠 **Alien Civilization Layers**  
+    ⭐ White = All Stars in Galaxy  
+    🔵 Blue = Stars with Planets ({planets})  
+    🟢 Green = Habitable Planets ({habitable})  
+    🟡 Yellow = Intelligent Life ({intelligent})  
+    🔴 Red = Communicative Civilizations ({communicative})  
+    📡 **Estimated Total Civilizations (N): {int(N)}**
+    """
 
-# Gradio UI
+    return fig, annotation
+
+# Gradio Interface
 interface = gr.Interface(
     fn=drake_simulator,
     inputs=[
@@ -58,9 +69,12 @@ interface = gr.Interface(
         gr.Slider(0.0, 1.0, value=0.5, label="fc (Communicative Fraction)"),
         gr.Slider(0.1, 10000, value=1000, label="L (Signal Lifetime)")
     ],
-    outputs=gr.Plot(),
+    outputs=[
+        gr.Plot(label="🪐 Galactic Simulation"),
+        gr.Markdown(label="📝 Explanation")
+    ],
     title="🛸 Alien Civilization Simulator",
-    description="Visualize the Drake Equation in Action 🚀"
+    description="Explore the Drake Equation in a galactic simulation 🌌. Adjust the values and watch the cosmic dance of life possibilities."
 )
 
 interface.launch()