Update app.py

app.py

@@ -55,22 +55,19 @@ def read_input_file(file):
     return df
 
 def query_openai_api(prompt):
-    import os
-    st.write(f"OPENAI_API_KEY: {os.getenv('OPENAI_API_KEY')}")
-    time.sleep(3)
-
     try:
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model="gpt-4o",
-            messages=[{"role": "system", "content": "You are a helpful assistant."},
-                      {"role": "user", "content": prompt}],
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": prompt}
+            ],
             max_tokens=2500,
             temperature=0.1
         )
-        responses = [choice['message']['content'].strip() for choice in response['choices']]
-        return responses
+        return [response.choices[0].message.content.strip()]
     except Exception as e:
-        print(f"Error querying OpenAI API: {e}")
+        st.error(f"Error querying OpenAI API: {e}")
         return []
 
 def process_data(df):