Update app.py

app.py

@@ -3,27 +3,31 @@ import openai
 
 def main():
     st.title("AI Chatbot")
-    
+
     # Input field for OpenAI API Key
     api_key = st.text_input("Enter your OpenAI API Key", type="password")
-    
+
     if api_key:
         openai.api_key = api_key
-        
         user_input = st.text_area("Ask me anything:")
-        
+
         if st.button("Generate Response"):
-            if user_input.strip() != "":
-                response = openai.ChatCompletion.create(
-                    model="gpt-3.5-turbo",
-                    messages=[{"role": "user", "content": user_input}]
-                )
-                
-                st.subheader("Response:")
-                st.write(response["choices"][0]["message"]["content"])
+            if user_input.strip():
+                try:
+                    response = openai.ChatCompletion.create(
+                        model="gpt-3.5-turbo",
+                        messages=[{"role": "user", "content": user_input}]
+                    )
+                    
+                    st.toast("Generating response...")  # Show a popup message
+                    st.subheader("Response:")
+                    st.write(response["choices"][0]["message"]["content"])
+                except Exception as e:
+                    st.error(f"Error: {str(e)}")  # Show error popup
             else:
                 st.warning("Please enter a question.")
     else:
         st.warning("Please enter your OpenAI API Key to proceed.")
 
-
+if __name__ == "__main__":
+    main()