Update app.py

app.py

@@ -3,23 +3,18 @@ from huggingface_hub import InferenceClient
 import PyPDF2
 import io
 from docx import Document
-import os
 
 # For PDF generation
 from reportlab.pdfgen import canvas
 from reportlab.lib.pagesizes import letter
-from reportlab.lib import utils
-from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer
-from reportlab.lib.styles import getSampleStyleSheet
 
 # Initialize the inference client from Hugging Face.
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-
-def extract_text_from_pdf(pdf_file):
-    """Extract text from PDF file."""
+def extract_text_from_pdf(pdf_file_bytes):
+    """Extract text from PDF bytes."""
     try:
-        pdf_reader = PyPDF2.PdfReader(pdf_file)
+        pdf_reader = PyPDF2.PdfReader(io.BytesIO(pdf_file_bytes))
         text = ""
         for page in pdf_reader.pages:
             page_text = page.extract_text()
@@ -29,40 +24,38 @@ def extract_text_from_pdf(pdf_file):
     except Exception as e:
         return f"Error reading PDF: {e}"
 
-
-def extract_text_from_docx(docx_file):
-    """Extract text from DOCX file."""
+def extract_text_from_docx(docx_file_bytes):
+    """Extract text from DOCX bytes."""
     try:
-        doc = Document(docx_file)
+        doc = Document(io.BytesIO(docx_file_bytes))
         text = "\n".join(para.text for para in doc.paragraphs)
         return text.strip() or "No text could be extracted from the DOCX file."
     except Exception as e:
         return f"Error reading DOCX: {e}"
 
-
-def parse_cv(file, job_description):
+def parse_cv(file_bytes, job_description):
     """Analyze the CV, show the prompt (debug) and return LLM analysis."""
-    if file is None:
+    if file_bytes is None:
         return "Please upload a CV file.", ""
 
+    # Determine file extension by header bytes
     try:
-        file_path = file.name  # Get the file path
-        file_ext = os.path.splitext(file_path)[1].lower()
-
-        if file_ext == ".pdf":
-            extracted_text = extract_text_from_pdf(file_path)
-        elif file_ext == ".docx":
-            extracted_text = extract_text_from_docx(file_path)
+        if file_bytes.startswith(b'%PDF'):
+            file_ext = "pdf"
+        elif file_bytes.startswith(b'PK\x03\x04'):
+            file_ext = "docx"
         else:
-            return (
-                "Unsupported file format. Please upload a PDF or DOCX file.",
-                "Unsupported file format.",
-            )
-
+            return "Unsupported file format.", "Cannot determine file type from content"
     except Exception as e:
         error_msg = f"Error reading file: {e}"
         return error_msg, error_msg
 
+    # Extract text
+    if file_ext == "pdf":
+        extracted_text = extract_text_from_pdf(file_bytes)
+    else:  # docx
+        extracted_text = extract_text_from_docx(file_bytes)
+
     # Check for extraction errors
     if extracted_text.startswith("Error"):
         return extracted_text, "Error during text extraction. Please check the file."
@@ -87,15 +80,7 @@ def parse_cv(file, job_description):
     except Exception as e:
         return extracted_text, f"Analysis Error: {e}"
 
-
-def respond(
-    message,
-    history: list[tuple[str, str]],
-    system_message,
-    max_tokens,
-    temperature,
-    top_p,
-):
+def respond(message, history: list[tuple[str, str]], system_message, max_tokens, temperature, top_p):
     """Generate chatbot response."""
     messages = [{"role": "system", "content": system_message}]
     for user_msg, bot_msg in history:
@@ -120,38 +105,32 @@ def respond(
     except Exception as e:
         yield f"Error during chat generation: {e}"
 
-
 def create_pdf_report(report_text):
-    """Creates a PDF report using SimpleDocTemplate for better formatting."""
+    """Creates a PDF report and returns it as (bytes, filename)."""
     if not report_text.strip():
         report_text = "No analysis report to convert."
 
-    buffer = io.BytesIO()
-    doc = SimpleDocTemplate(buffer, pagesize=letter)
-    styles = getSampleStyleSheet()
-    Story = []
-
-    # Title
-    Story.append(Paragraph("<b>Analysis Report</b>", styles["Title"]))
-    Story.append(Spacer(1, 12))
+    pdf_buffer = io.BytesIO()
+    c = canvas.Canvas(pdf_buffer, pagesize=letter)
+    c.setFont("Helvetica-Bold", 14)
+    c.drawString(72, 750, "Analysis Report")
 
-    # Report Content
+    text_obj = c.beginText(72, 730)
+    text_obj.setFont("Helvetica", 11)
     for line in report_text.split("\n"):
-        Story.append(Paragraph(line, styles["Normal"]))
-        Story.append(Spacer(1, 6))  # Add a small space between lines
+        text_obj.textLine(line)
+    c.drawText(text_obj)
 
-    doc.build(Story)
-    buffer.seek(0)
-    return buffer
+    c.showPage()
+    c.save()
+    pdf_buffer.seek(0)
 
+    # Return as a bytes-filename tuple that Gradio will treat as a file download
+    return (pdf_buffer.getvalue(), "analysis_report.pdf")
 
 def toggle_download_button(analysis_report):
-    """Toggle the download button."""
-    return gr.update(
-        interactive=bool(analysis_report.strip()),
-        visible=bool(analysis_report.strip()),
-    )
-
+    """Toggle the download button if we have an analysis report."""
+    return gr.update(interactive=bool(analysis_report.strip()), visible=bool(analysis_report.strip()))
 
 # Build the Gradio UI
 demo = gr.Blocks()
@@ -159,40 +138,30 @@ with demo:
     gr.Markdown("## AI-powered CV Analyzer and Chatbot")
 
     with gr.Tab("Chatbot"):
+        # Simple chat interface
         chat_interface = gr.ChatInterface(
             respond,
-            chatbot=gr.Chatbot(value=[], label="Chatbot"),
-            textbox=gr.Textbox(placeholder="Enter your message here...", label="Message"),
+            chatbot=gr.Chatbot(value=[], label="Chatbot", type="messages"),
+            type="messages",
             additional_inputs=[
-                gr.Textbox(
-                    value="You are a friendly Chatbot.", label="System message"
-                ),
-                gr.Slider(
-                    minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"
-                ),
-                gr.Slider(
-                    minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"
-                ),
-                gr.Slider(
-                    minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"
-                ),
+                gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
+                gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+                gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
+                gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
             ],
         )
 
     with gr.Tab("CV Analyzer"):
         gr.Markdown("### Upload your CV and provide the job description")
-        file_input = gr.File(label="Upload CV", file_types=[".pdf", ".docx"])
+        # IMPORTANT: set type="bytes" so we get raw bytes in parse_cv
+        file_input = gr.File(label="Upload CV", type="bytes")
         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", interactive=False)
+
+        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")
 
@@ -207,8 +176,10 @@ with demo:
         )
 
         download_pdf_button.click(
-            create_pdf_report, inputs=[analysis_output], outputs=[pdf_file]
+            create_pdf_report,
+            inputs=[analysis_output],
+            outputs=[pdf_file],
         )
 
 if __name__ == "__main__":
-    demo.queue().launch()
+    demo.queue().launch()