Update app.py

app.py

@@ -2,6 +2,8 @@ import os
 import json
 import base64
 import io
+import ast
+import logging
 from abc import ABC, abstractmethod
 from typing import Dict, List, Optional, Any
 
@@ -25,9 +27,8 @@ from Bio import Entrez  # Ensure BioPython is installed
 
 from dotenv import load_dotenv
 import requests
-import openai  # Updated for OpenAI SDK v1.8.0+
-import ast
-import logging
+import openai  # Updated for OpenAI SDK v1.0.0+
+from openai.error import APIError, RateLimitError, InvalidRequestError
 
 # ---------------------- Load Environment Variables ---------------------------
 load_dotenv()
@@ -44,11 +45,7 @@ logger = logging.getLogger()
 # ---------------------- Streamlit Page Configuration ---------------------------
 st.set_page_config(page_title="AI Clinical Intelligence Hub", layout="wide")
 
-# ---------------------- Import OpenAIError ---------------------------
-from openai.error import OpenAIError
-
-# ---------------------- Initialize External Clients ---------------------------
-# Initialize OpenAI API details from environment variables
+# ---------------------- Initialize OpenAI Client ---------------------------
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 PUB_EMAIL = os.getenv("PUB_EMAIL", "")
 
@@ -56,10 +53,12 @@ if not OPENAI_API_KEY:
     st.error("OpenAI API key must be set as an environment variable (OPENAI_API_KEY).")
     st.stop()
 
-# Set the OpenAI API key directly
-openai.api_key = OPENAI_API_KEY
+# Instantiate the OpenAI client
+client = openai.OpenAI(
+    api_key=OPENAI_API_KEY
+)
 
-# Load spaCy model with error handling
+# ---------------------- Load spaCy Model ---------------------------
 try:
     nlp = spacy.load("en_core_web_sm")
 except OSError:
@@ -495,7 +494,7 @@ class SimpleMedicalKnowledge(MedicalKnowledgeBase):
             """
 
             # Make the API request to OpenAI GPT-4
-            response = openai.ChatCompletion.create(
+            response = client.chat.completions.create(
                 model="gpt-4",
                 messages=[
                     {"role": "system", "content": "You are a helpful medical assistant."},
@@ -516,7 +515,13 @@ class SimpleMedicalKnowledge(MedicalKnowledgeBase):
             # Format the response
             return f"**Based on your query:** {answer}\n\n**PubMed Abstract:**\n\n{pubmed_abstract}"
 
-        except OpenAIError as e:
+        except RateLimitError as e:
+            logger.error(f"Rate Limit Exceeded: {str(e)}")
+            return "Rate limit exceeded. Please try again later."
+        except InvalidRequestError as e:
+            logger.error(f"Invalid Request: {str(e)}")
+            return f"Invalid request: {str(e)}"
+        except APIError as e:
             logger.error(f"OpenAI API Error: {str(e)}")
             return f"OpenAI API Error: {str(e)}"
         except Exception as e: