Update app.py
Browse files
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.
|
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
|
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:
|