Update app.py

app.py

@@ -107,28 +107,16 @@ def respond(message, history: list[tuple[str, str]], system_message, max_tokens,
         yield f"Error during chat generation: {e}"
 
 def create_pdf_report(report_text):
-    """
-    Creates a PDF from the given report text and returns a dictionary:
-      {
-        "name": "analysis_report.pdf",
-        "data": bytes_of_pdf,
-        "mime_type": "application/pdf"
-      }
-    which gr.File can accept as output.
-    """
+    """Creates a PDF report."""
     if not report_text.strip():
-        # If there's no report, just return a basic PDF stating so
         report_text = "No analysis report to convert."
 
-    # Use ReportLab to generate PDF in-memory
     pdf_buffer = io.BytesIO()
     c = canvas.Canvas(pdf_buffer, pagesize=letter)
 
-    # Title
     c.setFont("Helvetica-Bold", 14)
     c.drawString(72, 750, "Analysis Report")
 
-    # Body
     text_obj = c.beginText(72, 730)
     text_obj.setFont("Helvetica", 11)
     for line in report_text.split("\n"):
@@ -146,16 +134,8 @@ def create_pdf_report(report_text):
     }
 
 def toggle_download_button(analysis_report):
-    """
-    Enable the download PDF button only if there's a non-empty report.
-    """
-    if analysis_report.strip():
-        # Enable and make it visible
-        return gr.update(interactive=True, visible=True)
-    else:
-        # Hide and disable
-        return gr.update(interactive=False, visible=False)
-
+    """Toggle the download button."""
+    return gr.update(interactive=bool(analysis_report.strip()), visible=bool(analysis_report.strip()))
 
 # Build the Gradio UI
 demo = gr.Blocks()
@@ -181,8 +161,11 @@ with demo:
         job_desc_input = gr.Textbox(label="Job Description", lines=5)
         extracted_text = gr.Textbox(label="Extracted CV Content", lines=10, interactive=False)
         analysis_output = gr.Textbox(label="Analysis Report", lines=10, interactive=False)
+        download_pdf_button = gr.Button("Download Analysis as PDF", visible=False, interactive=False)
+        pdf_file = gr.File(label="Download PDF", file_count="single", interactive=False)
 
         analyze_button = gr.Button("Analyze CV")
+
         analyze_button.click(
             parse_cv,
             inputs=[file_input, job_desc_input],
@@ -190,25 +173,14 @@ with demo:
         ).then(
             toggle_download_button,
             inputs=[analysis_output],
-            outputs=[]  # we'll update the download button below
-        )
-
-        # Button to generate/download PDF
-        download_pdf_button = gr.Button("Download Analysis as PDF", visible=False, interactive=False)
-        pdf_file = gr.File(label="Download PDF", file_count="single", interactive=False)
-
-        # Chain a .then() so that once parse_cv finishes, we toggle the PDF button
-        analyze_button.then(
-            fn=toggle_download_button,
-            inputs=[analysis_output],
             outputs=[download_pdf_button]
         )
 
         download_pdf_button.click(
             create_pdf_report,
-            inputs=[analysis_output],  # pass the analysis text
-            outputs=[pdf_file]         # this will allow downloading the PDF
+            inputs=[analysis_output],
+            outputs=[pdf_file]
         )
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.queue().launch()