Update app.py

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 import openai
+from openai.error import OpenAIError
 import os
 
 # Securely fetch the OpenAI API key
@@ -45,9 +46,10 @@ def recommend_ai_model_via_gpt(description):
         )
         recommendation = response['choices'][0]['message']['content'].strip()
         return recommendation
-    except openai.error.OpenAIError as e:
+    except OpenAIError as e:
         return f"Error: {e}"
 
+
 def explain_recommendation(model_name):
     messages = [
         {"role": "user", "content": f"Why would {model_name} be a suitable choice for the application?"}
@@ -60,7 +62,7 @@ def explain_recommendation(model_name):
         )
         explanation = response['choices'][0]['message']['content'].strip()
         return explanation
-    except openai.error.OpenAIError as e:
+    except OpenAIError as e:
         return f"Error: {e}"
 
 # Streamlit UI