import streamlit as st
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():
                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()