mgbam commited on
Commit
e58975b
·
verified ·
1 Parent(s): 12ec03d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -25,7 +25,7 @@ from Bio import Entrez # Ensure BioPython is installed
25
 
26
  from dotenv import load_dotenv
27
  import requests
28
- import openai # Updated for OpenAI SDK v1.0
29
  import ast
30
  import logging
31
 
@@ -44,6 +44,9 @@ logger = logging.getLogger()
44
  # ---------------------- Streamlit Page Configuration ---------------------------
45
  st.set_page_config(page_title="AI Clinical Intelligence Hub", layout="wide")
46
 
 
 
 
47
  # ---------------------- Initialize External Clients ---------------------------
48
  # Initialize OpenAI API details from environment variables
49
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
@@ -492,7 +495,7 @@ class SimpleMedicalKnowledge(MedicalKnowledgeBase):
492
  """
493
 
494
  # Make the API request to OpenAI GPT-4
495
- response = self.client.chat_completions.create(
496
  model="gpt-4",
497
  messages=[
498
  {"role": "system", "content": "You are a helpful medical assistant."},
@@ -513,7 +516,7 @@ class SimpleMedicalKnowledge(MedicalKnowledgeBase):
513
  # Format the response
514
  return f"**Based on your query:** {answer}\n\n**PubMed Abstract:**\n\n{pubmed_abstract}"
515
 
516
- except openai.error.OpenAIError as e:
517
  logger.error(f"OpenAI API Error: {str(e)}")
518
  return f"OpenAI API Error: {str(e)}"
519
  except Exception as e:
 
25
 
26
  from dotenv import load_dotenv
27
  import requests
28
+ import openai # Updated for OpenAI SDK v1.8.0+
29
  import ast
30
  import logging
31
 
 
44
  # ---------------------- Streamlit Page Configuration ---------------------------
45
  st.set_page_config(page_title="AI Clinical Intelligence Hub", layout="wide")
46
 
47
+ # ---------------------- Import OpenAIError ---------------------------
48
+ from openai.error import OpenAIError
49
+
50
  # ---------------------- Initialize External Clients ---------------------------
51
  # Initialize OpenAI API details from environment variables
52
  OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
495
  """
496
 
497
  # Make the API request to OpenAI GPT-4
498
+ response = self.client.chat.completions.create(
499
  model="gpt-4",
500
  messages=[
501
  {"role": "system", "content": "You are a helpful medical assistant."},
 
516
  # Format the response
517
  return f"**Based on your query:** {answer}\n\n**PubMed Abstract:**\n\n{pubmed_abstract}"
518
 
519
+ except OpenAIError as e:
520
  logger.error(f"OpenAI API Error: {str(e)}")
521
  return f"OpenAI API Error: {str(e)}"
522
  except Exception as e: