Update app.py

app.py

@@ -27,20 +27,10 @@ LOSSDOG_PROMPT = """
 """
 
 def extract_text_from_file(file_path: str, file_name: str) -> str:
-    """
-    Extract text from a file. Handles PDF and plain text files.
-    Args:
-        file_path (str): The path to the file.
-        file_name (str): The name of the file.
-    Returns:
-        str: Extracted text content from the file.
-    """
     if file_name.endswith(".pdf"):
         try:
             pdf_reader = PdfReader(file_path)
-            text = ""
-            for page in pdf_reader.pages:
-                text += page.extract_text()
+            text = "".join([page.extract_text() for page in pdf_reader.pages])
             return text
         except Exception as e:
             return f"Error extracting text from PDF: {str(e)}"
@@ -54,31 +44,15 @@ def extract_text_from_file(file_path: str, file_name: str) -> str:
         return "Unsupported file format. Please upload a PDF or TXT file."
 
 def convert_to_markdown(text: str) -> str:
-    """
-    Convert plain text to Markdown format for better readability.
-    Args:
-        text (str): The plain text to convert.
-    Returns:
-        str: The Markdown-formatted text.
-    """
     return markdownify(text, heading_style="ATX")
 
 def analyze_resume_with_lossdog(
     markdown_text: str, api_key: str, history: list
 ) -> list:
-    """
-    Analyze the resume using GPT-4o-mini and generate conversational responses.
-    Args:
-        markdown_text (str): Markdown content of the resume.
-        api_key (str): OpenAI API key.
-        history (list): Chat history between the user and the assistant.
-    Returns:
-        list: Updated chat history with AI responses.
-    """
     try:
         openai.api_key = api_key
         messages = [
-            {"role": "system", "content": LOSSDOG_PROMPT},
+            {"role": "system", "content": LOSS DOG_PROMPT},
             {"role": "system", "content": f"Resume Content:\n{markdown_text}"},
         ] + history
 
@@ -95,13 +69,6 @@ def analyze_resume_with_lossdog(
         return history
 
 def generate_json_output(history: list) -> tuple:
-    """
-    Generate a JSON summary of the chat history and save it to a file.
-    Args:
-        history (list): Chat history between the user and the assistant.
-    Returns:
-        tuple: File name and content of the generated JSON file.
-    """
     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
     filename = f"resume_analysis_{timestamp}.json"
     try:
@@ -113,9 +80,6 @@ def generate_json_output(history: list) -> tuple:
         return None, f"Error generating JSON: {str(e)}"
 
 def create_demo():
-    """
-    Create the Gradio Blocks interface for the Loss Dog resume analyzer.
-    """
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("""
         # 🐕 LOSS Dog: AI-Powered Resume Analyzer
@@ -137,7 +101,7 @@ def create_demo():
             upload = gr.File(label="Upload Your Resume (PDF or TXT)")
 
         markdown_preview = gr.Markdown(label="Resume Content (Markdown)")
-        chatbot = gr.Chatbot(label="Chat with LOSS Dog")
+        chatbot = gr.Chatbot(label="Chat with LOSS Dog", type="messages")
 
         with gr.Row():
             user_input = gr.Textbox(
@@ -148,17 +112,13 @@ def create_demo():
 
         with gr.Row():
             json_view = gr.JSON(label="Final JSON Summary")
-            download_button = gr.File(label="Download JSON Summary")
+            download_link = gr.Textbox(label="Download Link (Copy & Paste)")
 
-        # Internal states for the Gradio app
         history_state = gr.State([])
         markdown_state = gr.State("")
         file_path_state = gr.State("")
 
         def handle_upload(file, api_key):
-            """
-            Handle resume file upload. Convert to Markdown and update chat history.
-            """
             if not file:
                 return None, "No file uploaded.", [], None
 
@@ -177,48 +137,38 @@ def create_demo():
             return file, markdown_text, history, markdown_text
 
         def handle_message(message, api_key, history, markdown_text):
-            """
-            Process user messages and generate Loss Dog's response.
-            """
             if not message.strip():
                 return history
 
             return analyze_resume_with_lossdog(markdown_text, api_key, history)
 
         def handle_generate_json(history):
-            """
-            Generate JSON summary from the conversation history.
-            """
             filename, json_content = generate_json_output(history)
             if not filename:
-                return None, json_content
-            return json_content, filename
+                return json_content, ""
+            download_url = f"/download/{filename}"
+            return json_content, download_url
 
-        # Event: When a file is uploaded
         upload.change(
             handle_upload,
             inputs=[upload, api_key],
             outputs=[upload, markdown_preview, chatbot, markdown_state]
         )
 
-        # Event: When a message is sent
         send_button.click(
             handle_message,
             inputs=[user_input, api_key, history_state, markdown_state],
             outputs=[chatbot]
         )
 
-        # Event: Generate and download JSON
-        download_button.click(
+        send_button.click(
             handle_generate_json,
             inputs=[history_state],
-            outputs=[json_view, download_button]
+            outputs=[json_view, download_link]
         )
 
     return demo
 
-
-# Main execution
 if __name__ == "__main__":
     demo = create_demo()
     demo.launch()