Update app.py

app.py

@@ -12,7 +12,7 @@ import json
 from rank_bm25 import BM25Okapi
 import pandas as pd
 import gradio as gr
-from openai import OpenAI
+import openai
 
 # Load dataset
 dataset_url = "https://huggingface.co/datasets/username/mental-health-classification/resolve/main/train.csv"
@@ -27,7 +27,8 @@ if "OPENAI_API_KEY" not in os.environ:
     raise ValueError("Please set your OpenAI API key using `os.environ['OPENAI_API_KEY'] = 'your_api_key'`")
 
 # Initialize OpenAI API
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
 
 # Few-shot classification function
 def classify_text(input_text, k=20):
@@ -65,7 +66,7 @@ def classify_text(input_text, k=20):
     """
 
     try:
-        response = client.chat.completions.create(
+        response = openai.ChatCompletion.create(
             messages=[
                 {"role": "system", "content": "You are a mental health specialist."},
                 {"role": "user", "content": prompt},