Update app.py

app.py

@@ -1,24 +1,20 @@
 import gradio as gr
-from timetable import generate_full_timetable
-from pdf_utils import export_pdf
+from timetable import generate_timetable
+from pdf_utils import export_timetable_pdf
 
-COURSES_FILE = "courses.csv"
-ROOMS_FILE = "rooms.csv"
-
-def generate():
-    df = generate_full_timetable(COURSES_FILE, ROOMS_FILE)
-    pdf_path = export_pdf(df)
-    return df, pdf_path
+def generate_and_export(courses_file, rooms_file):
+    timetable_dict = generate_timetable(courses_file.name, rooms_file.name)
+    pdf_file = export_timetable_pdf(timetable_dict)
+    return pdf_file
 
 with gr.Blocks() as demo:
-    gr.Markdown("## 📅 Multi-Class/Program Weekly Time Table Generator")
-
-    btn = gr.Button("Generate Time Table")
-    table = gr.Dataframe(label="Generated Time Table")
-    pdf = gr.File(label="Download PDF")
-
-    btn.click(fn=generate, outputs=[table, pdf])
-
-    gr.Markdown("🖨️ **Print Tip:** Download PDF → Open → Print")
+    gr.Markdown("## Class/Program Wise Time Table Generator")
+    with gr.Row():
+        courses_file = gr.File(label="Upload courses.csv")
+        rooms_file = gr.File(label="Upload rooms.csv")
+    generate_btn = gr.Button("Generate Time Table")
+    pdf_output = gr.File(label="Download PDF")
+    
+    generate_btn.click(generate_and_export, inputs=[courses_file, rooms_file], outputs=[pdf_output])
 
 demo.launch()