Update app.py

app.py

@@ -1,14 +1,14 @@
 import os
 import gradio as gr
 import pandas as pd
-import openai
+from openai import OpenAI
 from dotenv import load_dotenv
 
 # Load environment variables
 load_dotenv()
 
-# Set up OpenAI API key
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Set up OpenAI client
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 # Simple database using pandas DataFrames
 class SimpleDatabase:
@@ -83,7 +83,7 @@ class QueryRouter:
     def _classify_query(self, query):
         """Classify the query to determine which agent should handle it"""
         # Use OpenAI to classify the query
-        response = openai.chat.completions.create(
+        response = client.chat.completions.create(
             model="gpt-3.5-turbo",
             messages=[
                 {"role": "system", "content": """
@@ -122,7 +122,7 @@ class QueryRouter:
         Example: {{"product_name": "laptop"}} or {{"date": "2025-04-29"}}
         """
         
-        response = openai.chat.completions.create(
+        response = client.chat.completions.create(
             model="gpt-3.5-turbo",
             messages=[
                 {"role": "system", "content": "You extract parameters from customer queries for a shop assistant."},
@@ -142,7 +142,7 @@ class QueryRouter:
             
     def _handle_general_knowledge(self, query):
         """Handle general knowledge queries using OpenAI"""
-        response = openai.chat.completions.create(
+        response = client.chat.completions.create(
             model="gpt-3.5-turbo",
             messages=[
                 {"role": "system", "content": """