Update app.py

app.py

@@ -28,7 +28,7 @@ from Bio import Entrez  # Ensure BioPython is installed
 from dotenv import load_dotenv
 import requests
 import openai  # Updated for OpenAI SDK v1.0.0+
-from openai import OpenAIError, RateLimitError, BadRequestError
+from openai.exceptions import OpenAIError, RateLimitError, BadRequestError
 
 # ---------------------- Load Environment Variables ---------------------------
 load_dotenv()
@@ -56,6 +56,11 @@ if not OPENAI_API_KEY:
 # Set the OpenAI API key
 openai.api_key = OPENAI_API_KEY
 
+# Update OpenAI base_url and proxies if needed
+# Example:
+# openai.api_base = "https://api.openai.com/v1"
+# openai.proxies = {"https": "http://proxy.example.com:8080"}
+
 # ---------------------- Load spaCy Model ---------------------------
 try:
     nlp = spacy.load("en_core_web_sm")
@@ -516,10 +521,10 @@ class SimpleMedicalKnowledge(MedicalKnowledgeBase):
         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:
+        except BadRequestError as e:
+            logger.error(f"Bad Request: {str(e)}")
+            return f"Bad request: {str(e)}"
+        except OpenAIError as e:
             logger.error(f"OpenAI API Error: {str(e)}")
             return f"OpenAI API Error: {str(e)}"
         except Exception as e: