Update app.py

app.py

@@ -1,36 +1,11 @@
 import gradio as gr
 import openai
-import os
 from PyPDF2 import PdfReader
 from markdownify import markdownify
 
-# Enhanced system prompt
-LOSSDOG_PROMPT = """
-<LossDogFramework version="3.0">
-    <Identity>
-        <Description>
-            You are Loss Dog, a state-of-the-art AI-powered career advisor, resume analyzer, and builder. Your primary role is to:
-            - Carefully read and analyze the user’s resume.
-            - Use the resume as a knowledge context for all interactions.
-            - Proactively engage with the user by answering questions, identifying areas of improvement, and offering suggestions.
-            - Help the user build a stronger, more tailored resume by focusing on content clarity, relevance, and impact.
-        </Description>
-    </Identity>
-    <CoreDirectives>
-        <Mission>
-            Your mission is to collaborate with users to improve their resumes and provide actionable advice. To achieve this, you must:
-            - Use the uploaded resume as a knowledge base.
-            - Answer user questions or provide feedback based on the resume content.
-            - Offer actionable suggestions and best practices.
-        </Mission>
-    </CoreDirectives>
-</LossDogFramework>
-"""
-
-def extract_text_from_file(file_path: str, file_name: str) -> str:
-    """
-    Extract text from a file. Handles PDF and plain text files.
-    """
+LOSSDOG_PROMPT = "SYSTEM PROMPT: You are Loss Dog, a resume analyzer..."
+
+def extract_text_from_file(file_path, file_name):
     if file_name.endswith(".pdf"):
         try:
             pdf_reader = PdfReader(file_path)
@@ -47,136 +22,111 @@ def extract_text_from_file(file_path: str, file_name: str) -> str:
     else:
         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.
-    """
+def convert_to_markdown(text):
     return markdownify(text, heading_style="ATX")
 
-def interact_with_lossdog(user_message: str, markdown_text: str, api_key: str, history: list) -> list:
-    """
-    Send a message to LOSS Dog with the resume as persistent context.
-    """
+def interact_with_lossdog(user_message, markdown_text, api_key, history):
+    # Ensure openai.api_key is set
+    openai.api_key = api_key
+
+    # Validate existing history to keep only well-formed messages
+    validated_history = []
+    for msg in history:
+        if isinstance(msg, dict) and "role" in msg and "content" in msg:
+            validated_history.append({"role": msg["role"], "content": msg["content"]})
+
+    # Build the system + resume context
+    messages = [
+        {"role": "system", "content": LOSSDOG_PROMPT},
+        {"role": "system", "content": f"Resume Content:\n{markdown_text}"}
+    ] + validated_history
+
+    # Add the new user message
+    messages.append({"role": "user", "content": user_message})
+
     try:
-        openai.api_key = api_key
-
-        # Validate and format the history
-        formatted_history = [
-            {"role": entry["role"], "content": entry["content"]}
-            for entry in history
-            if isinstance(entry, dict) and "role" in entry and "content" in entry
-        ]
-
-        # Build the conversation messages
-        messages = [
-            {"role": "system", "content": LOSSDOG_PROMPT},
-            {"role": "system", "content": f"Resume Content:\n{markdown_text}"}
-        ] + formatted_history + [{"role": "user", "content": user_message}]
-
-        # Generate a response
         response = openai.ChatCompletion.create(
-            model="gpt-4",
+            model="gpt-4o-mini",
             messages=messages,
-            max_tokens=500,
+            max_tokens=500
         )
-        assistant_response = response.choices[0].message.content
-
-        # Update history
-        history.append({"role": "user", "content": user_message})
-        history.append({"role": "assistant", "content": assistant_response})
-        return history
+        assistant_reply = response.choices[0].message.content
+        # Update local history
+        validated_history.append({"role": "user", "content": user_message})
+        validated_history.append({"role": "assistant", "content": assistant_reply})
     except Exception as e:
-        history.append({"role": "assistant", "content": f"Error: {str(e)}"})
-        return history
+        validated_history.append({"role": "assistant", "content": f"Error: {e}"})
+
+    return validated_history
 
 def create_demo():
-    """
-    Create the Gradio Blocks interface for the Loss Dog resume analyzer.
-    """
     with gr.Blocks(css="#resume-preview { height: 300px; overflow-y: auto; border: 1px solid #ccc; padding: 10px; }") as demo:
-        gr.Markdown("""
-        # 🐕 LOSS Dog: AI-Powered Resume Advisor
-        
-        Upload your resume (PDF or TXT), and Loss Dog will:
-        - Display it in a scrollable box on the top-right.
-        - Use it as persistent context for all interactions.
-        - Engage in back-and-forth chat to help improve your resume and answer questions.
-        """)
-
-        # API Key Input
+        gr.Markdown("# 🐕 LOSS Dog: AI-Powered Resume Advisor")
+
         api_key = gr.Textbox(
             label="OpenAI API Key",
             placeholder="Enter your OpenAI API key...",
             type="password"
         )
 
-        # 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 (Scrollable)",
-                    elem_id="resume-preview"
-                )
+                markdown_preview = gr.Markdown(label="Resume Preview", elem_id="resume-preview")
 
-        # User input and send button
         with gr.Row():
-            user_input = gr.Textbox(
-                placeholder="Ask LOSS Dog about your resume...",
-                label="Your Message",
-                lines=1
-            )
+            user_input = gr.Textbox(label="Your Message", lines=1)
             send_button = gr.Button("Send 🐾")
 
-        # File upload
         with gr.Row():
             upload = gr.File(label="Upload Your Resume (PDF or TXT)")
 
-        # App state
+        # States
         history_state = gr.State([])
         markdown_state = gr.State("")
 
+        # 1) File Upload Handler
         def handle_upload(file, api_key):
             """
-            Handle resume upload. Convert it to Markdown and set as context.
+            Convert file to Markdown & reset/return history as empty or minimal.
             """
             if not file:
+                # If no file uploaded, show error in the preview & return empty chat
                 return "No file uploaded.", []
-
-            file_path = file.name
-            file_content = extract_text_from_file(file_path, file.name)
-            if "Error" in file_content:
-                return file_content, []
-
-            # Convert resume to Markdown for preview
-            markdown_text = convert_to_markdown(file_content)
-            return markdown_text, []
-
+            text = extract_text_from_file(file.name, file.name)
+            if text.startswith("Error"):
+                # Show error in preview & return empty chat
+                return text, []
+            # Convert to Markdown for display
+            resume_md = convert_to_markdown(text)
+            # On upload, we might start fresh or keep existing chat? Let's reset to empty
+            return resume_md, []
+
+        # 2) Chat Handler
         def handle_message(user_message, api_key, markdown_text, history):
             """
-            Handle back-and-forth chat with the user.
+            Perform actual chat with stored resume context & return updated history.
             """
-            return interact_with_lossdog(user_message, markdown_text, api_key, history)
+            updated_history = interact_with_lossdog(user_message, markdown_text, api_key, history)
+            # Return two outputs: one for chatbot, one for saving state
+            return updated_history, updated_history
 
-        # File upload event
+        # Connect events
         upload.change(
-            handle_upload,
+            fn=handle_upload,
             inputs=[upload, api_key],
             outputs=[markdown_preview, history_state]
         )
 
-        # Chat message event
         send_button.click(
-            handle_message,
+            fn=handle_message,
             inputs=[user_input, api_key, markdown_state, history_state],
             outputs=[chatbot, history_state]
         )
 
     return demo
 
-
-# Main execution
 if __name__ == "__main__":
     demo = create_demo()
     demo.launch()