Update app.py

app.py

@@ -11,8 +11,7 @@ import io
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Keep your existing CONVERSATION_PROMPT and EXTRACTION_PROMPT here
-
+# Prompts
 CONVERSATION_PROMPT = """You are LOSS DOG, a professional profile builder. Your goal is to have natural conversations
 with users to gather information about their professional background across 9 categories:
 
@@ -137,42 +136,47 @@ Return the data in this exact structure:
 }
 
 IMPORTANT: Return ONLY the JSON. Do not add any explanation text."""
+
 class ProfileBuilder:
     def __init__(self):
-        self.conversation_history = []
         self.client = None
         self.pdf_text = None
     
     def _initialize_client(self, api_key: str) -> None:
         """Initialize OpenAI client if not already initialized"""
-        if not self.client:
-            if not api_key.startswith("sk-"):
-                raise ValueError("Invalid API key format")
-            self.client = AsyncOpenAI(api_key=api_key)
+        if not api_key.startswith("sk-"):
+            raise ValueError("Invalid API key format")
+        self.client = AsyncOpenAI(api_key=api_key)
 
-    async def process_message(self, message: str, api_key: str) -> Dict[str, Any]:
-        """Process a chat message"""
+    async def process_message(self, message: str, history: List[List[str]], api_key: str) -> Dict[str, Any]:
+        """Process a chat message using conversation history from Gradio's state"""
         try:
             # Initialize client if needed
             self._initialize_client(api_key)
-
-            # Add user message to history
-            self.conversation_history.append({"role": "user", "content": message})
+            
+            # Convert Gradio history format to OpenAI message format
+            conversation_history = []
+            for human, assistant in history:
+                conversation_history.extend([
+                    {"role": "user", "content": human},
+                    {"role": "assistant", "content": assistant}
+                ])
+            
+            # Add current message
+            conversation_history.append({"role": "user", "content": message})
             
             # Get AI response
             completion = await self.client.chat.completions.create(
-                model="gpt-4o-mini",
+                model="gpt-4-0125-preview",
                 messages=[
                     {"role": "system", "content": CONVERSATION_PROMPT},
-                    *self.conversation_history
+                    *conversation_history
                 ],
                 temperature=0.7
             )
             
-            # Extract and store response
+            # Extract response
             ai_message = completion.choices[0].message.content
-            self.conversation_history.append({"role": "assistant", "content": ai_message})
-            
             return {"response": ai_message}
 
         except Exception as e:
@@ -187,7 +191,7 @@ class ProfileBuilder:
             text = ""
             for page in pdf_reader.pages:
                 text += page.extract_text()
-            self.pdf_text = text  # Store the PDF text
+            self.pdf_text = text
             return text
         except Exception as e:
             logger.error(f"PDF extraction error: {str(e)}")
@@ -204,7 +208,7 @@ class ProfileBuilder:
 
             # Process with AI
             completion = await self.client.chat.completions.create(
-                model="gpt-4o-mini",
+                model="gpt-4-0125-preview",
                 messages=[
                     {"role": "system", "content": EXTRACTION_PROMPT},
                     {"role": "user", "content": f"Extract profile information from this resume:\n\n{resume_text}"}
@@ -231,18 +235,14 @@ class ProfileBuilder:
             logger.error(f"PDF processing error: {str(e)}")
             return {"error": str(e)}
 
-    async def generate_profile(self) -> tuple[Dict[str, Any], Optional[str]]:
+    async def generate_profile(self, history: List[List[str]], api_key: str) -> tuple[Dict[str, Any], Optional[str]]:
         """Generate profile from conversation or PDF"""
         try:
-            if not self.client:
-                raise ValueError("OpenAI client not initialized")
+            self._initialize_client(api_key)
 
             # Determine source and prepare content
-            if self.conversation_history:
-                content = "\n".join(
-                    f"{msg['role']}: {msg['content']}"
-                    for msg in self.conversation_history
-                )
+            if history:
+                content = "\n".join(f"User: {msg[0]}\nAssistant: {msg[1]}" for msg in history)
                 source = "conversation"
             elif self.pdf_text:
                 content = self.pdf_text
@@ -252,7 +252,7 @@ class ProfileBuilder:
 
             # Get AI extraction
             completion = await self.client.chat.completions.create(
-                model="gpt-4o-mini",
+                model="gpt-4-0125-preview",
                 messages=[
                     {"role": "system", "content": EXTRACTION_PROMPT},
                     {"role": "user", "content": f"Extract profile information from this {source}:\n\n{content}"}
@@ -340,7 +340,7 @@ def create_gradio_interface():
             if not message.strip():
                 return history, None, None
 
-            result = await builder.process_message(message, key)
+            result = await builder.process_message(message, history, key)
             
             if "error" in result:
                 return history, {"error": result["error"]}, None
@@ -367,8 +367,8 @@ def create_gradio_interface():
             except Exception as e:
                 return {"error": str(e)}, None
 
-        async def on_generate():
-            profile, filename = await builder.generate_profile()
+        async def on_generate(history: List[List[str]], key: str):
+            profile, filename = await builder.generate_profile(history, key)
             if "error" in profile:
                 return {"error": profile["error"]}, None
             return profile["profile_data"], filename
@@ -377,13 +377,13 @@ def create_gradio_interface():
         msg.submit(
             on_message,
             inputs=[msg, chatbot, api_key],
-            outputs=[chatbot, profile_output, msg]
+            outputs=[chatbot, profile_output, download_btn]
         )
         
         send.click(
             on_message,
             inputs=[msg, chatbot, api_key],
-            outputs=[chatbot, profile_output, msg]
+            outputs=[chatbot, profile_output, download_btn]
         )
         
         process_pdf_btn.click(
@@ -394,6 +394,7 @@ def create_gradio_interface():
         
         generate_btn.click(
             on_generate,
+            inputs=[chatbot, api_key],
             outputs=[profile_output, download_btn]
         )