Updated OpenAI API and fixed async batch processing

app.py

@@ -5,31 +5,32 @@ import textstat
 import os
 import asyncio
 
-# Load API key from Hugging Face secrets
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Initialize OpenAI client
+client = openai.OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 # Function to fetch available OpenAI models
 def get_models():
-    models = openai.Model.list()
-    return [model['id'] for model in models["data"]]
+    models = client.models.list()
+    return [model.id for model in models.data]
 
 # Function to process AI response
 def generate_response(prompt, model, tone):
-    response = openai.ChatCompletion.create(
+    response = client.chat.completions.create(
         model=model,
         messages=[{"role": "system", "content": f"Rewrite this in {tone} style: {prompt}"}]
     )
-    return response["choices"][0]["message"]["content"].strip()
+    return response.choices[0].message.content.strip()
 
 # Function for batch processing with async calls
 async def process_bulk(prompts, model, tone):
     tasks = [
-        openai.ChatCompletion.acreate(
+        client.chat.completions.acreate(
             model=model,
             messages=[{"role": "system", "content": f"Rewrite this in {tone} style: {p}"}]
         ) for p in prompts
     ]
-    return await asyncio.gather(*tasks)
+    responses = await asyncio.gather(*tasks)
+    return [response.choices[0].message.content.strip() for response in responses]
 
 # UI Structure
 st.title("AI Prompt Optimizer")
@@ -83,7 +84,7 @@ with st.expander("📂 Upload CSV for Bulk Optimization"):
         if "Prompt" in df.columns:
             prompts = df["Prompt"].tolist()
             optimized_prompts = asyncio.run(process_bulk(prompts, model_choice, tone_choice))
-            df["Optimized_Prompt"] = [p["choices"][0]["message"]["content"].strip() for p in optimized_prompts]
+            df["Optimized_Prompt"] = optimized_prompts
             st.write(df)
             st.download_button("Download Optimized CSV", df.to_csv(index=False).encode('utf-8'), "optimized_prompts.csv", "text/csv")
         else: