Update app.py

app.py

@@ -1012,8 +1012,7 @@ DATASET = [
     {'class_id': 'C0999', 'name': 'Finance 401 — Valuation', 'professor': 'Jamie Wright', 'days': 'Thu,Fri', 'times': '9:15 AM - 10:45 AM', 'subject': 'Finance'},
     {'class_id': 'C1000', 'name': 'Finance 101 — Behavioral Finance: Foundations', 'professor': 'Parker Anderson', 'days': 'Mon,Fri', 'times': '4:15 PM - 6:15 PM', 'subject': 'Finance'},
 ]
-
-df = pd.DataFrame(DATASET)
+  df = pd.DataFrame(DATASET)
 
 # ---------- CONSTANTS ----------
 SUBJECTS = sorted(df["subject"].unique().tolist())
@@ -1130,8 +1129,17 @@ def pick_schedules(df_pool: pd.DataFrame, demand: Dict[str, int], max_attempts=5
 def draw_timetable(schedule_rows: List[dict], title: str):
     fig, ax = plt.subplots(figsize=(10, 7), dpi=150)
     ax.set_xlim(0, 7); ax.set_ylim(8, 21)
-    ax.set_xticks(range(7)); ax.set_xticklabels(DAYS_AXIS)
-    ax.set_yticks(range(8, 22, 1)); ax.set_ylabel("Time"); ax.set_title(title)
+
+    # Put days on TOP, not bottom
+    ax.set_xticks(range(7))
+    ax.set_xticklabels(DAYS_AXIS)
+    ax.xaxis.tick_top()
+    ax.xaxis.set_label_position('top')
+    ax.tick_params(axis='x', which='both', bottom=False, top=True, labelbottom=False, labeltop=True)
+
+    ax.set_yticks(range(8, 22, 1))
+    ax.set_ylabel("Time")
+    ax.set_title(title, pad=20)
 
     for x in range(8): ax.axvline(x, linewidth=0.5)
     for y in range(8, 22): ax.axhline(y, linewidth=0.3)
@@ -1203,7 +1211,8 @@ def generate(tbl, instructions):
     tbl = _ensure_array_table(tbl)
     if not tbl:
         blank = draw_timetable([], "Schedule option 1 (empty)")
-        return blank, "Schedule option 1", [], 0, gr.Button.update(visible=False), gr.Button.update(visible=False), gr.Button.update(visible=False), gr.update(visible=False)
+        # hide nav + details when empty
+        return blank, "Schedule option 1", [], 0, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
     demand = {}
     for subject, count in tbl:
@@ -1213,21 +1222,35 @@ def generate(tbl, instructions):
 
     if not demand:
         blank = draw_timetable([], "Schedule option 1 (empty)")
-        return blank, "Schedule option 1", [], 0, gr.Button.update(visible=False), gr.Button.update(visible=False), gr.Button.update(visible=False), gr.update(visible=False)
+        return blank, "Schedule option 1", [], 0, gr.update(visible=False), gr.update(visible=False), gr.update(visible=False), gr.update(visible=False)
 
     pool = filter_by_constraints(df, instructions)
     scheds = pick_schedules(pool, demand)
+
+    # Always render something; if algo couldn't fit, you'll just see a blank grid
     idx = 0
     title = f"Schedule option {idx+1}"
-    img = draw_timetable(scheds[idx], title)
-    return img, title, scheds, idx, gr.Button.update(visible=False), gr.Button.update(visible=True if len(scheds)>1 else False), gr.Button.update(visible=True), gr.update(visible=False)
+    img = draw_timetable(scheds[idx] if idx < len(scheds) else [], title)
+
+    # Show nav (if >1) and details button
+    prev_vis = gr.update(visible=False)  # first page: prev hidden
+    next_vis = gr.update(visible=True if len(scheds) > 1 else False)
+    details_vis = gr.update(visible=True)
+    hide_details_table = gr.update(visible=False)  # reset details table until clicked again
+
+    return img, title, scheds, idx, prev_vis, next_vis, details_vis, hide_details_table
 
 def step(direction, scheds, idx):
-    if not scheds: return gr.update(), "", idx, gr.Button.update(visible=False), gr.Button.update(visible=False)
-    idx = (idx + 1) % len(scheds) if direction=="next" else (idx - 1) % len(scheds)
+    if not scheds:
+        return gr.update(), "", idx, gr.update(visible=False), gr.update(visible=False)
+    if direction == "next":
+        idx = (idx + 1) % len(scheds)
+    else:
+        idx = (idx - 1) % len(scheds)
     title = f"Schedule option {idx+1}"
     img = draw_timetable(scheds[idx], title)
-    return img, title, idx, gr.Button.update(visible=True), gr.Button.update(visible=True)
+    # once you start navigating, show both buttons
+    return img, title, idx, gr.update(visible=True), gr.update(visible=True)
 
 def get_details(scheds, idx):
     rows = scheds[idx] if (scheds and 0 <= idx < len(scheds)) else []
@@ -1252,8 +1275,8 @@ with gr.Blocks(css="""
             subject_table = gr.Dataframe(
                 headers=["Subject","Count"],
                 datatype=["str","number"],
-                type="array",             # ✅ returns list-of-lists
-                value=[],                 # ✅ start empty
+                type="array",             # returns list-of-lists
+                value=[],
                 row_count=(0,"dynamic"),
                 col_count=2,
                 interactive=True,
@@ -1297,9 +1320,7 @@ with gr.Blocks(css="""
     subject_table.change(update_total, inputs=[subject_table], outputs=[total_text])
 
     def _gen(tbl, instr):
-        img, title, scheds, idx, pvis, nvis, dvis, hide_details = generate(tbl, instr)
-        return img, title, scheds, idx, pvis, nvis, dvis, hide_details
-
+        return generate(tbl, instr)
     generate_btn.click(
         _gen,
         inputs=[subject_table, custom_instructions],