Update app.py

app.py

@@ -1,14 +1,16 @@
+
 import json
 import logging
 from datetime import datetime
 from typing import Dict, List, Optional, Any
 import gradio as gr
-from openai import OpenAI  # Updated import
+from openai import AsyncOpenAI  # Changed to AsyncOpenAI
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# System prompts remain the same as before
 # System 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:
@@ -49,10 +51,10 @@ class ProfileBuilder:
         self.client = None
     
     def _initialize_client(self, api_key: str) -> None:
-        """Initialize OpenAI client with API key."""
+        """Initialize AsyncOpenAI client with API key."""
         if not api_key.startswith("sk-"):
             raise ValueError("Invalid API key format")
-        self.client = OpenAI(api_key=api_key)
+        self.client = AsyncOpenAI(api_key=api_key)
     
     async def process_message(self, message: str, api_key: str) -> Dict[str, Any]:
         """Process a user message through conversation phase."""
@@ -63,8 +65,8 @@ class ProfileBuilder:
             # Add message to history
             self.conversation_history.append({"role": "user", "content": message})
             
-            # Get AI response using new client format
-            response = await self.client.chat.completions.create(
+            # Get AI response - properly awaited
+            completion = await self.client.chat.completions.create(
                 model="gpt-4o-mini",
                 messages=[
                     {"role": "system", "content": CONVERSATION_PROMPT},
@@ -73,7 +75,7 @@ class ProfileBuilder:
                 temperature=0.7
             )
             
-            ai_message = response.choices[0].message.content
+            ai_message = completion.choices[0].message.content
             self.conversation_history.append({"role": "assistant", "content": ai_message})
             
             return {"response": ai_message}
@@ -94,8 +96,8 @@ class ProfileBuilder:
                 for msg in self.conversation_history
             )
 
-            # Extract structured information using new client format
-            response = await self.client.chat.completions.create(
+            # Extract structured information - properly awaited
+            completion = await self.client.chat.completions.create(
                 model="gpt-4o-mini",
                 messages=[
                     {"role": "system", "content": EXTRACTION_PROMPT},
@@ -105,7 +107,7 @@ class ProfileBuilder:
             )
 
             # Parse the structured response
-            profile_data = json.loads(response.choices[0].message.content)
+            profile_data = json.loads(completion.choices[0].message.content)
 
             # Add metadata
             profile = {
@@ -201,4 +203,8 @@ def create_gradio_interface():
 
 if __name__ == "__main__":
     demo = create_gradio_interface()
-    demo.launch(server_name="0.0.0.0", server_port=7860)
+    demo.queue()  # Add queue for async support
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=7860
+    )