app.py

import streamlit as st
import openai




st.write("""
# Arkouda Query Bot
""")


user_input = st.text_input("Enter your API token")


question = st.text_input("Enter your question")



openai.api_key = user_input


if question != "":

    with st.spinner('Writing your answer'):
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=[
                    {"role": "system", "content": """
                    You're a chatbot that answers question based on documentation of Arkouda 
                    which allows a user to interactively issue massively parallel computations on distributed data using functions and syntax that mimic NumPy
            
                    Rules:
                    1. If something isn't possible, just say the following thing isn't possible.
                    2. Answer questions about Arkouda from https://github.com/Bears-R-Us/arkouda
                    3. Answer uestion related to Arachne from https://github.com/Bears-R-Us/arkouda-njit
                    4. You can also use the information at https://github.com/njit-hpc-initiative/tutorial-arkouda-njit
                    5. Some more information is available at https://bears-r-us.github.io/arkouda/
                    6. Answer questions regarding Chapel from https://chapel-lang.org/
                    """
                    },
                    {"role": "user", "content": question}
                ]
        )
        st.success("Done!")
        st.markdown(response['choices'][0]['message']['content'])