add chatbot.py, main.py, and moderator.py

data_science_tutor/chatbot.py

@@ -0,0 +1,59 @@
+from openai import OpenAI
+import streamlit as st
+from moderator import moderator
+
+
+
+gpt_model = 'gpt-3.5-turbo'
+client = OpenAI(
+    api_key= "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"
+)
+
+messages = [
+    {
+        'role' : "system",
+        'content' : "You are a data science tutor who provides short, simple, clear answers.\
+                     You can't provide information on a wide range of topics beyond data science and programming."
+    },
+    {
+        'role' : 'user',
+        'content' : 'Tell me about yourself.'
+    },
+    {
+        'role' : 'assistant',
+        'content' : "I am a data sciece assistant. My purpose is to assist users by providing concise and\
+                     accurate answers in the topics related to data science, programming, mathematics,\
+                    Artificial Intelligence,Machine Learning, Deep Learning and related fields." 
+    },
+    {
+        'role' : 'user',
+        'content' : 'Can you provide information in topics not related to data science and programming?'
+    },
+    {
+        'role' : 'assistant',
+        'content' : "No, I'm here to assist yout in data related fields such data science, programming,\
+                     mathematics and more." 
+    }   
+]
+
+
+# GPT-3.5 response generation
+def chatbot(prompt:str):  
+    flagged = moderator(prompt)
+    st.write('User: ', prompt)
+    if flagged == True:
+        st.write("Assistant: ", "The latest question violates OpenAI's content policy. Please change it.", "\n")
+
+    else:
+        user_dict = {'role' : 'user', 'content' : prompt}
+        messages.append(user_dict)
+
+        res = client.chat.completions.create(
+            model=gpt_model,
+            messages=messages
+        )
+        assistant_dict = {'role' : res.choices[0].message.role, 'content' : res.choices[0].message.content}
+        messages.append(assistant_dict)
+        st.write("Assistant: ", assistant_dict['content'], "\n")
+    
+

data_science_tutor/main.py

@@ -0,0 +1,21 @@
+import streamlit as st
+from chatbot import chatbot
+
+
+
+# Streamlit application
+def combined_model():  
+    st.title("Data Science Assistant")
+    curr_question = st.text_area(
+        'enter your prompt',
+        placeholder="Talk to your Data Science Tutor: ",
+        label_visibility="hidden"
+    )
+    if st.button("Generate Answer"):
+        try:
+            chatbot(curr_question)
+        except:
+            st.warning(body="Refresh the page or Try it again later.", icon="🤖")
+
+combined_model()
+

data_science_tutor/moderator.py

@@ -0,0 +1,17 @@
+from openai import OpenAI
+
+
+
+moderation_model = 'text-moderation-latest'
+client = OpenAI(
+    api_key="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"
+)
+
+
+def moderator(prompt:str):
+    response = client.moderations.create(
+        input= prompt,
+        model= moderation_model,
+    )
+
+    return response.results[0].flagged