Update app.py

app.py

@@ -38,48 +38,33 @@ messages = [
 ]
 
 
-def moderator(prompt:str):
-    response = client.moderations.create(
-        input= prompt,
-        model= moderation_model,
-    )
-
-    return response.results[0].flagged
-
-
-# GPT-3.5 response generation
-def chatbot(prompt:str):  
-    flagged = moderator(prompt)
-    st.write('User: ', prompt)
-    if flagged == True:
-        st.warning("Assistant: The latest request violates OpenAI's content policies. 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
+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:
+        response = client.moderations.create(
+            input= curr_question,
+            model= moderation_model,
         )
-        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")
-
-
-# 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="🤖")
+        flagged = response.results[0].flagged
+        st.write('User: ', curr_question)
+        if flagged == True:
+            st.warning("Assistant: The latest request violates OpenAI's content policies. Please change it.\n",  "🚨")
 
+        else:
+            user_dict = {'role' : 'user', 'content' : curr_question}
+            messages.append(user_dict)
 
-combined_model()
+            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")
+    except:
+        st.warning(body="Refresh the page or Try it again later.", icon="🤖")