Update app.py

app.py

@@ -53,6 +53,9 @@ with st.sidebar:
             Enjoy chatting with Meta's Llama3 model!
         """)
 
+    # Text input
+    user_topic = st.text_input("Enter a topic", "Data Science")
+
     # Add a button to submit
     submit_button = st.button("Submit")
 
@@ -76,24 +79,70 @@ for message in st.session_state.messages:
         st.markdown(message["content"])
 
 
+# Create agents
+interviewer = call_llama
+interviewee = call_llama
+judge = call_llama
+
+
 # React to user input
+iter = 0
 if submit_button:
 
     # Initiatization
-    prompt = "Ask a data science question"
+    prompt = f"Ask a question about this topic: {user_topic}"
 
     # Display user message in chat message container
     st.chat_message("user").markdown(prompt)
-
-    # Add user message to chat history
     st.session_state.messages.append({"role": "user", "content": prompt})
 
-    # API call to Llama
-    response = call_llama(prompt)
-
-    # Display assistant response in chat message container
-    with st.chat_message("assistant"):
-        st.markdown(response)
+    while True:
+
+        # Interview asks a question
+        question = interviewer(prompt)
+
+        # Display assistant response in chat message container
+        st.chat_message("assistant").markdown(question)
+        st.session_state.messages.append({"role": "assistant", "content": question})
+
+        # Interviewee attempts an answer
+        if iter < 5:
+            answer = interviewee(
+                f"""
+                    Answer the question: {question} in a mediocre way
+
+                    Because you are an inexperienced interviewee.
+                """
+            )
+            st.chat_message("user").markdown(prompt)
+            st.session_state.messages.append({"role": "user", "content": prompt})
+        else:
+            answer = interviewee(
+                f"""
+                    Answer the question: {question} in a mediocre way
+
+                    Because you are an inexperienced interviewee but you really want to learn,
+                    so you learn from the judge comments: {judge_comments}
+                """
+            )
+            st.chat_message("user").markdown(prompt)
+            st.session_state.messages.append({"role": "user", "content": prompt})
+
+        # Judge thinks and advises but the thoughts are hidden
+        judge_comments = judge(
+            f"""
+                The question is: {question}
+
+                The answer is: {answer}
+
+                Provide feedback and rate the answer from 1 to 10 while 10 being the best and 1 is the worst. 
+            """
+        )
+        
+        # Stopping rule
+        if '9' in judge_comments:
+            break
+
+        # Checkpoint
+        iter += 1
 
-    # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response})