Update app.py

app.py

@@ -3,7 +3,7 @@ import logging
 from datetime import datetime
 from typing import Dict, List, Optional, Any
 import gradio as gr
-import openai
+from openai import OpenAI  # Updated import
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -43,24 +43,28 @@ structured information into the following categories:
 
 Format the output as clean, structured JSON. Include confidence scores for each extracted piece of information.
 Mark any inferred information clearly."""
-
 class ProfileBuilder:
     def __init__(self):
         self.conversation_history = []
-        self.api_key = None
+        self.client = None
+    
+    def _initialize_client(self, api_key: str) -> None:
+        """Initialize OpenAI client with API key."""
+        if not api_key.startswith("sk-"):
+            raise ValueError("Invalid API key format")
+        self.client = OpenAI(api_key=api_key)
     
     async def process_message(self, message: str, api_key: str) -> Dict[str, Any]:
         """Process a user message through conversation phase."""
         try:
-            if not self.api_key:
-                self.api_key = api_key
-                openai.api_key = api_key
+            if not self.client:
+                self._initialize_client(api_key)
 
             # Add message to history
             self.conversation_history.append({"role": "user", "content": message})
             
-            # Get AI response
-            response = await openai.ChatCompletion.acreate(
+            # Get AI response using new client format
+            response = await self.client.chat.completions.create(
                 model="gpt-4o-mini",
                 messages=[
                     {"role": "system", "content": CONVERSATION_PROMPT},
@@ -81,14 +85,17 @@ class ProfileBuilder:
     async def generate_profile(self) -> Dict[str, Any]:
         """Process conversation history into structured profile."""
         try:
+            if not self.client:
+                raise ValueError("OpenAI client not initialized")
+
             # Convert conversation history to text
             conversation_text = "\n".join(
                 f"{msg['role']}: {msg['content']}"
                 for msg in self.conversation_history
             )
 
-            # Extract structured information
-            response = await openai.ChatCompletion.acreate(
+            # Extract structured information using new client format
+            response = await self.client.chat.completions.create(
                 model="gpt-4o-mini",
                 messages=[
                     {"role": "system", "content": EXTRACTION_PROMPT},
@@ -194,4 +201,4 @@ def create_gradio_interface():
 
 if __name__ == "__main__":
     demo = create_gradio_interface()
-    demo.launch(server_name="0.0.0.0", server_port=7860) 
+    demo.launch(server_name="0.0.0.0", server_port=7860)