Update app.py

app.py

@@ -15,7 +15,13 @@ def process_input(job_description, *resumes):
         formatted_output += f"Resume #{i}:\nScore: {score:.2f}\nResume Snippet: {resume[:200]}...\n\n-------\n\n"
     return formatted_output
 
-# UI definition
+
+def add_resume_count(*resumes):
+    resumes = [r for r in resumes if r and r.strip() != ""]
+    resumes.append("")
+    return resumes
+    
+
 with gr.Blocks() as demo:
     gr.Markdown("## 📝 Resume Ranking System")
 
@@ -23,18 +29,20 @@ with gr.Blocks() as demo:
 
     with gr.Row():
 
-        resume_count = gr.State(2)
+        # resume_count = gr.State(2)
+        resume_count = gr.State(["", ""])
 
         with gr.Column():
             @gr.render(inputs=resume_count)
             def render_count(count):
                 # print("[COUNT]", count)
                 # print("[RESUME LIST]", resumes_list)
-                for i in range(count):
+                for i in range(len(count)):
                     resume_input = gr.Textbox(
                         lines=6,
                         placeholder=f"Paste resume #{i+1} here...",
-                        label=f"Resume #{i+1}"
+                        label=f"Resume #{i+1}",
+                        value=count[i]
                     )
                     resumes_list.append(resume_input)
 
@@ -43,6 +51,7 @@ with gr.Blocks() as demo:
                     inputs=[job_description] + resumes_list,
                     outputs=output
                 )
+                add_resume_btn.click(add_resume_count, resumes_list, resume_count)
 
             # @gr.render(inputs=input_text)
             # def add_resume():
@@ -55,7 +64,7 @@ with gr.Blocks() as demo:
             #     return resumes_group.update(visible=True)
     
             add_resume_btn = gr.Button("➕ Add Another Resume")
-            add_resume_btn.click(lambda x: x + 1, resume_count, resume_count)
+            # add_resume_btn.click(lambda x: x + 1, resume_count, resume_count)
 
         with gr.Column():
             job_description = gr.Textbox(