Update app.py

app.py

@@ -27,10 +27,18 @@ 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 = "".join([page.extract_text() for page in pdf_reader.pages])
+            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)}"
@@ -44,22 +52,38 @@ 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 OpenAI GPT 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": LOSS DOG PROMPT},
+            {"role": "system", "content": LOSSDOG_PROMPT},
             {"role": "system", "content": f"Resume Content:\n{markdown_text}"},
         ] + history
 
         response = openai.ChatCompletion.create(
             model="gpt-4o-mini",
             messages=messages,
-            max_tokens=500,
+            max_tokens=50000,
         )
         assistant_response = response.choices[0].message.content
         history.append({"role": "assistant", "content": assistant_response})
@@ -69,6 +93,13 @@ 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:
@@ -80,6 +111,9 @@ 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
@@ -112,11 +146,9 @@ def create_demo():
 
         with gr.Row():
             json_view = gr.JSON(label="Final JSON Summary")
-            download_link = gr.Textbox(label="Download Link (Copy & Paste)")
 
         history_state = gr.State([])
         markdown_state = gr.State("")
-        file_path_state = gr.State("")
 
         def handle_upload(file, api_key):
             if not file:
@@ -144,10 +176,7 @@ def create_demo():
 
         def handle_generate_json(history):
             filename, json_content = generate_json_output(history)
-            if not filename:
-                return json_content, ""
-            download_url = f"/download/{filename}"
-            return json_content, download_url
+            return json_content
 
         upload.change(
             handle_upload,
@@ -161,14 +190,10 @@ def create_demo():
             outputs=[chatbot]
         )
 
-        send_button.click(
-            handle_generate_json,
-            inputs=[history_state],
-            outputs=[json_view, download_link]
-        )
-
     return demo
 
+
+# Main execution
 if __name__ == "__main__":
     demo = create_demo()
     demo.launch()