Update app.py

app.py

@@ -42,11 +42,6 @@ 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:
@@ -67,24 +62,12 @@ def extract_text_from_file(file_path: str, file_name: str) -> str:
 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:
+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
@@ -94,7 +77,7 @@ def analyze_resume_with_lossdog(
         ] + history
 
         response = openai.ChatCompletion.create(
-            model="gpt-4",
+            model="gpt-4o-mini",
             messages=messages,
             max_tokens=4096,  # Maximum token limit for GPT-4
         )
@@ -105,23 +88,6 @@ def analyze_resume_with_lossdog(
         history.append({"role": "assistant", "content": f"Error: {str(e)}"})
         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:
-        output_data = {"conversation_history": history}
-        with open(filename, "w", encoding="utf-8") as f:
-            json.dump(output_data, f, indent=2)
-        return filename, json.dumps(output_data, f, indent=2)
-    except Exception as e:
-        return None, f"Error generating JSON: {str(e)}"
 def create_demo():
     """
     Create the Gradio Blocks interface for the Loss Dog resume analyzer.
@@ -131,32 +97,41 @@ def create_demo():
         # 🐕 LOSS Dog: AI-Powered Resume Analyzer and Builder
         
         Upload your resume (PDF or TXT), and Loss Dog will:
-        - Display it in a box in the top-right corner.
-        - Proactively analyze it and start a conversation.
-        - Help you build a polished, professional resume with actionable feedback.
+        - Display your resume in the top-right box.
+        - Proactively analyze it and provide actionable feedback.
+        - Help you build a professional and tailored resume.
         """)
 
-        # Main layout: Left for chat, top-right for resume preview
-        with gr.Row():
-            chatbot = gr.Chatbot(label="Chat with LOSS Dog", type="messages")
+        # API Key Input
+        api_key = gr.Textbox(
+            label="OpenAI API Key",
+            placeholder="Enter your OpenAI API key...",
+            type="password",
+            show_label=True
+        )
 
-            # Small top-right container for resume preview
+        # Main Layout: Chatbot on the left, resume preview on the right
+        with gr.Row():
+            with gr.Column(scale=3):
+                chatbot = gr.Chatbot(label="Chat with LOSS Dog", type="messages")
             with gr.Column(scale=1):
-                markdown_preview = gr.Markdown(label="Resume Preview")  # Fixed: Removed interactive=False
+                markdown_preview = gr.Markdown(label="Resume Preview")
 
+        # User input and Send button
         with gr.Row():
             user_input = gr.Textbox(
                 placeholder="Ask LOSS Dog about your resume...",
-                label="Your Message"
+                label="Your Message",
+                lines=1
             )
             send_button = gr.Button("Send 🐾")
 
-        # File upload and JSON generation in a separate row
+        # File upload and JSON summary
         with gr.Row():
             upload = gr.File(label="Upload Your Resume (PDF or TXT)")
             json_view = gr.JSON(label="Final JSON Summary")
 
-        # App state
+        # App State
         history_state = gr.State([])
         markdown_state = gr.State("")
 
@@ -164,18 +139,18 @@ def create_demo():
             if not file:
                 return "No file uploaded.", [], None
 
-            # Extract resume content from the uploaded file
+            # Extract resume content
             file_path = file.name
             file_content = extract_text_from_file(file_path, file.name)
             if "Error" in file_content:
                 return file_content, [], None
 
-            # Convert extracted content to Markdown for display
+            # Convert to Markdown
             markdown_text = convert_to_markdown(file_content)
             history = [
                 {
                     "role": "assistant",
-                    "content": "I've reviewed your resume. Let's dive into ways to improve it! Here are some initial thoughts:",
+                    "content": "I've reviewed your resume. Let's discuss improvements! Here are my thoughts:",
                 }
             ]
             history = analyze_resume_with_lossdog(markdown_text, api_key, history)