updated

app.py

@@ -16,9 +16,11 @@ def load_csv(file):
             return None, f"❌ CSV must contain {required_cols}"
 
         # Add Age Group
-        df["Age Group"] = pd.cut(df["Age"],
-                                 bins=[20, 25, 30, 35, 40],
-                                 labels=["21-25", "26-30", "31-35", "36-40"])
+        df["Age Group"] = pd.cut(
+            df["Age"],
+            bins=[20, 25, 30, 35, 40],
+            labels=["21-25", "26-30", "31-35", "36-40"]
+        )
         return df, "✅ File loaded successfully!"
     except Exception as e:
         return None, f"❌ Error: {e}"
@@ -48,10 +50,11 @@ def salary_chart():
     plt.xticks(rotation=45)
     plt.grid(axis="y", linestyle="--", alpha=0.7)
     plt.tight_layout()
-    buf = io.BytesIO()
-    plt.savefig(buf, format="png")
-    buf.seek(0)
-    return buf
+
+    filepath = "salary_chart.png"
+    plt.savefig(filepath)
+    plt.close()
+    return filepath
 
 
 def performance_chart():
@@ -65,10 +68,11 @@ def performance_chart():
     plt.xticks(rotation=45)
     plt.grid(True, linestyle="--", alpha=0.6)
     plt.tight_layout()
-    buf = io.BytesIO()
-    plt.savefig(buf, format="png")
-    buf.seek(0)
-    return buf
+
+    filepath = "performance_chart.png"
+    plt.savefig(filepath)
+    plt.close()
+    return filepath
 
 
 def search_employee(query):
@@ -98,8 +102,9 @@ with gr.Blocks() as demo:
         salary_btn = gr.Button("💰 Salary Chart")
         performance_btn = gr.Button("🏆 Performance Chart")
 
-    salary_plot = gr.Image(type="auto", label="Salary Chart")
-    performance_plot = gr.Image(type="auto", label="Performance Chart")
+    # ✅ Fixed: use "filepath"
+    salary_plot = gr.Image(type="filepath", label="Salary Chart")
+    performance_plot = gr.Image(type="filepath", label="Performance Chart")
 
     # Events
     file_input.change(load_csv, inputs=file_input, outputs=[data_table, status])