Update main.py

main.py

@@ -6,8 +6,10 @@ from pydantic import BaseModel
 from datetime import datetime
 import uuid
 import os
-import openai
-openai.api_key = os.environ.get("OPENAI_API_KEY")
+from openai import OpenAI
+
+client = OpenAI()  # Automatically picks up your Hugging Face secret OPENAI_API_KEY
+
 from transformers import pipeline
 import logging, traceback
 from typing import Optional, List, Union
@@ -278,7 +280,7 @@ async def root_cause_analysis(payload: dict, x_api_key: str = Header(None)):
         Suggestion: ...
         """
 
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model="gpt-4",
             messages=[{"role": "user", "content": prompt}]
         )