Create app.py

app.py

@@ -0,0 +1,98 @@
+import gradio as gr
+from parser import parse_resume
+import json
+
+def process_input(job_description, *resumes):
+    resumes = [r for r in resumes if r and r.strip() != ""]  # Remove empty
+    if not job_description.strip() or not resumes:
+        return "Please provide both job description and at least one resume."
+
+    thinking, results = parse_resume(job_description, resumes)
+    # formatted_output = f""
+    return thinking, json.loads(results)
+    
+    # results = zip(*parse_resume(job_description, resumes))
+    # formatted_output = ""
+    # for i, (resume, score) in enumerate(results, 1):
+    #     formatted_output += f"Resume #{i}:\nScore: {score:.2f}\nResume Snippet: {resume[:200]}...\n\n-------\n\n"
+    # return formatted_output
+
+# UI definition
+with gr.Blocks() as demo:
+    gr.Markdown("## 📝 Resume Ranking System")
+
+    resumes_list = []
+
+    with gr.Row():
+
+        resume_count = gr.State(2)
+
+        with gr.Column():
+            @gr.render(inputs=resume_count)
+            def render_count(count):
+                for i in range(count):
+                    name = gr.Textbox(
+                        lines=1,
+                        placeholder=f"Category #{i+1} name",
+                        label=f"Name #{i+1}"
+                    )
+                    data_type = gr.Textbox(
+                        lines=1,
+                        placeholder=f"Category #{i+1} data type",
+                        label=f"Data Type #{i+1}"
+                    )
+                    desc = gr.Textbox(
+                        lines=1,
+                        placeholder=f"Category #{i+1} description",
+                        label=f"Description #{i+1}"
+                    )
+                    resumes_list.append(
+                        {"name":name, "type":data_type, "description":desc}
+                    )
+
+                submit_btn.click(
+                    fn=process_input,
+                    inputs=[job_description] + resumes_list,
+                    outputs=[thinking_output, output]
+                )
+
+            # @gr.render(inputs=input_text)
+            # def add_resume():
+            #     new_input = gr.Textbox(
+            #         lines=6,
+            #         placeholder=f"Paste resume #{len(resumes_list)+1} here...",
+            #         label=f"Resume #{len(resumes_list)+1}"
+            #     )
+            #     resumes_list.append(new_input)
+            #     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)
+
+        with gr.Column():
+            job_description = gr.Textbox(
+                lines=8,
+                placeholder="Paste Resume here...",
+                label="Job Description"
+            )
+
+            # output = gr.Textbox(
+            #     lines=10,
+            #     label="JSON Result",
+            #     interactive=False
+            # )
+            output = gr.JSON(show_indices=True)
+            thinking_output = gr.Textbox(
+                lines=10,
+                label="Thinking Result",
+                interactive=False
+            )
+            
+            submit_btn = gr.Button("🚀 Rank Resumes")
+
+    
+    # add_resume_btn.click(add_resume, outputs=resumes_group)
+    # add_resume_btn.click(lambda x: x + 1, resume_count, resume_count)
+
+
+demo.launch()