Update app.py

app.py

@@ -88,14 +88,15 @@ def get_resume_response(prompt: str, api_key: str, model: str = "llama-3.3-70b",
     return response_string
 
 def process_resume(file, jd_string):
-    # Initialize MarkItDown
-    md = MarkItDown()
+    """
+    Process the uploaded resume and job description, optimize it, and return the result.
+    """
     try:
-        # Convert file to Markdown
-        original_md = md.convert(file.name)
+        # Convert file to Markdown using MarkItDown
+        original_md = md_converter.convert(file.name)
         original_text = original_md.text_content
 
-        # Save the original Markdown to a file
+        # Save the original resume in Markdown format
         original_file_path = "resumes/original_resume.md"
         with open(original_file_path, "w", encoding="utf-8") as f:
             f.write(original_text)
@@ -104,18 +105,19 @@ def process_resume(file, jd_string):
         prompt = create_prompt(original_text, jd_string)
         response_string = get_resume_response(prompt, api_key)
 
-        # Extract optimized resume and suggestions
+        # Split response into optimized resume and suggestions
         response_list = response_string.split("## Additional Suggestions")
-        new_resume = response_list[0]
+        new_resume = response_list[0].strip()
+        suggestions = "## Additional Suggestions\n\n" + response_list[1].strip() if len(response_list) > 1 else ""
 
-        # Save the optimized Markdown to a file
+        # Save the optimized resume in Markdown format
         optimized_file_path = "resumes/optimized_resume.md"
         with open(optimized_file_path, "w", encoding="utf-8") as f:
             f.write(new_resume)
 
-        return original_text, new_resume, "resumes/original_resume.md", "resumes/optimized_resume.md"
+        return original_text, new_resume, original_file_path, optimized_file_path, suggestions
     except Exception as e:
-        return f"Error processing file: {str(e)}", "", None, None
+        return f"Error processing file: {str(e)}", "", None, None, ""
 
 def export_resume(new_resume):
     try:
@@ -151,7 +153,11 @@ with gr.Blocks() as app:
     export_result = gr.Markdown(label="Export Result")
 
     # Bindings
-    run_button.click(process_resume, inputs=[resume_input, jd_input], outputs=[before_md, after_md, download_before, download_after, output_suggestions])
+    run_button.click(
+    process_resume,
+    inputs=[resume_input, jd_input],
+    outputs=[before_md, after_md, download_before, download_after, output_suggestions]
+    )
     export_button.click(export_resume, inputs=[after_md], outputs=[export_result])
 
 app.launch()