Update app.py

app.py

@@ -2,7 +2,8 @@ import gradio as gr
 import random
 import matplotlib.pyplot as plt
 import pandas as pd
-from io import StringIO
+import tempfile
+import os
 
 # -------- Generate Sample Data --------
 def generate_jobs(num_jobs=5, num_machines=3):
@@ -62,17 +63,10 @@ def plot_schedule(schedule, title):
     ax.set_yticks(range(max(m for _, m, _, _ in schedule) + 1))
     ax.set_yticklabels([f"Machine {m+1}" for m in range(max(m for _, m, _, _ in schedule) + 1)])
     plt.tight_layout()
+
     plt.close(fig)
     return fig
 
-# -------- Convert Schedule to CSV --------
-def schedule_to_csv(schedule):
-    df = pd.DataFrame(schedule, columns=["Job", "Machine", "Start Time", "End Time"])
-    csv_buffer = StringIO()
-    df.to_csv(csv_buffer, index=False)
-    csv_buffer.seek(0)
-    return csv_buffer.getvalue()
-
 # -------- Main Gradio Function --------
 def run_scheduler(num_jobs, num_machines, algorithm):
     jobs = generate_jobs(num_jobs, num_machines)
@@ -86,9 +80,16 @@ def run_scheduler(num_jobs, num_machines, algorithm):
     else:
         return None, "Invalid algorithm selected.", None
 
+    # Create Gantt Chart
     fig = plot_schedule(schedule, title)
-    csv_data = schedule_to_csv(schedule)
-    return fig, f"🔹 Makespan: {makespan} time units", csv_data
+
+    # Create a temporary CSV file
+    df = pd.DataFrame(schedule, columns=["Job", "Machine", "Start Time", "End Time"])
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".csv")
+    df.to_csv(temp_file.name, index=False)
+
+    # Return figure, performance metric, and download link
+    return fig, f"🔹 Makespan: {makespan} time units", temp_file.name
 
 # -------- Gradio Interface --------
 demo = gr.Interface(
@@ -101,10 +102,10 @@ demo = gr.Interface(
     outputs=[
         gr.Plot(label="Generated Schedule"),
         gr.Textbox(label="Performance Metric", interactive=False),
-        gr.File(label="Download Schedule (CSV)")
+        gr.File(label="Download Schedule CSV")
     ],
     title="🧠 Smart Factory Job Scheduling Visualizer",
-    description="Visualize job scheduling in smart factories, compare algorithms, and download the schedule as CSV."
+    description="Generate job schedules, visualize Gantt charts, and export scheduling data for analysis."
 )
 
 demo.launch()