Update app.py

app.py

@@ -19,8 +19,20 @@ def generate_response(prompt, model_name, api_key):
             messages=[{"role": "user", "content": prompt}]
         )
         return completion.choices[0].message.content
-    except openai.error.OpenAIError as e:
-        st.error(f"Error with {model_name}: {e}")
+    except openai.APIError as e:  # Corrected error handling
+        st.error(f"OpenAI API Error with {model_name}: {e}")
+        return None
+    except openai.RateLimitError as e:
+        st.error(f"OpenAI Rate Limit Error with {model_name}: {e}")
+        return None
+    except openai.AuthenticationError as e:
+        st.error(f"OpenAI Authentication Error with {model_name}: {e}")
+        return None
+    except openai.ServiceUnavailableError as e:
+        st.error(f"OpenAI Service Unavailable Error with {model_name}: {e}")
+        return None
+    except Exception as e:  # Catch any other exception
+        st.error(f"An unexpected error occurred with {model_name}: {e}")
         return None
 
 # Main Streamlit app