Update app.py

app.py

@@ -146,41 +146,31 @@ def find_jobs(file, added_kw, use_ai):
     explanation = refine_with_ai(ranked, resume) if use_ai else ""
     return table, explanation
 
-# def paginate(table, page):
-#     per_page = 10
-#     start = (page-1)*per_page
-#     return table[start:start+per_page]
-
 # 7️⃣ Jobs in Markdown
 def jobs_to_markdown(table, page, per_page=PER_PAGE):
     total = len(table)
     pages = max(1, math.ceil(total / per_page))
     page = max(1, min(page, pages))
-    start = (page - 1) * per_page
-    end   = start + per_page
-    slice = table[start:end]
+    start, end = (page-1)*per_page, page*per_page
+    slice_ = table[start:end]
 
-    # Header
     md  = f"**Showing jobs {start+1}–{min(end,total)} of {total} (Page {page}/{pages})**\n\n"
     md += "| Role | Company | Location | Posted | Score | Apply |\n"
     md += "| ---- | ------- | -------- | ------ | ----- | ----- |\n"
-    # Rows
-    for row in slice:
+    for row in slice_:
         link = f"[Apply]({row['Apply']})" if row['Apply'] else ""
         md += (
             f"| {row['Role']} | {row['Company']} | {row['Location']} "
             f"| {row['Posted']} | {row['Score']} | {link} |\n"
         )
-    # # Append explanation *only* on page 1
-    # if page == 1 and explanation:
-    #     md += "\n---\n**AI Explanation:**\n\n" + explanation
     return md
 
 def load_jobs_and_pages(resume, added_kw, use_ai):
-    full_table, explanation = find_jobs(resume, added_kw, use_ai)  # your existing pipeline
+    full_table, explanation = find_jobs(resume, added_kw, use_ai)
     total_pages = max(1, math.ceil(len(full_table) / PER_PAGE))
-    # Return: table-state, explanation-state, slider update
-    return full_table, explanation, gr.update(value=1, maximum=total_pages)
+    slider_update = gr.update(value=1, maximum=total_pages)
+    first_page_md = jobs_to_markdown(full_table, 1)
+    return full_table, explanation, slider_update, first_page_md
 
 # 7️⃣ Gradio UI
 with gr.Blocks(theme=gr.themes.Base()) as demo:
@@ -193,40 +183,26 @@ with gr.Blocks(theme=gr.themes.Base()) as demo:
     use_ai = gr.Checkbox(label="Use AI to refine explanation", value=False)
     find_btn = gr.Button("Find Jobs")
 
-    # hidden states for full data
-    jobs_state = gr.State([])
-    expl_state = gr.State("")
-
-    # outputs
-    jobs_md = gr.Markdown()
-    expl_md = gr.Markdown()
-    page_sel = gr.Slider(minimum=1, maximum=1, step=1, value=1, label="Page")
+    jobs_state = gr.State([])           # holds full table
+    page_sel   = gr.Slider(1, 1, step=1, value=1, label="Page")
+    jobs_md    = gr.Markdown()          # shows the current page’s markdown
+    expl_md_h  = gr.Markdown("### AI Explanation")
+    expl_md    = gr.Markdown()          # shows AI explanation
 
-    # 1) On search: populate states & slider
+    # 1) On Find Jobs:
+    #    - load the jobs, explanation, slider max, and first-page markdown
     find_btn.click(
         fn=load_jobs_and_pages,
         inputs=[resume, added, use_ai],
-        outputs=[jobs_state, expl_state, page_sel]
-    ).then(
-        fn=lambda tbl, pg: jobs_to_markdown(tbl, pg),
-        inputs=[jobs_state, page_sel],
-        outputs=[jobs_md]
+        outputs=[jobs_state, expl_md, page_sel, jobs_md]
     )
 
-    # 2) On page change: re-render table only
+    # 2) On page change, re-render only the table markdown
     page_sel.change(
         fn=lambda tbl, pg: jobs_to_markdown(tbl, pg),
         inputs=[jobs_state, page_sel],
         outputs=[jobs_md]
     )
 
-    # 3) Always render explanation from state
-    expl_state.change(
-        fn=lambda expl: expl,
-        inputs=[expl_state],
-        outputs=[expl_md],
-        queue=False
-    )
-
 if __name__ == "__main__":
     demo.launch()