teacher replies

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 from ollama import chat
+from loguru import logger
+
 
 available_models = [
     'openhermes',
@@ -13,16 +15,18 @@ available_models = [
 
 
 def ask(model, system_prompt, pre_prompt, question):
-    response = chat(model=model, messages=[
+    messages = [
         {
             'role': 'system',
-            'content': f"{system_prompt}${pre_prompt}",
+            'content': f"{system_prompt} {pre_prompt}",
         },
         {
             'role': 'user',
-            'content': f'${question}',
+            'content': f"{question}",
         },
-    ])
+    ]
+    logger.debug(f"<< {model} << {messages}")
+    response = chat(model=model, messages=messages)
     return response['message']['content']
 
 
@@ -31,8 +35,8 @@ def main():
     sp2 = """There are 3 people standing in a circle: the Priest, the Teacher (that's you) and the Kid."""
     sp3 = """There are 3 people standing in a circle: the Priest, the Teacher and the Kid (that's you)."""
 
-    pp1 = pp2 = pp3 = "Ask the other two by always starting your sentence with their role. Always share your inner thoughts inside parentheses."
-    qp1 = qp2 = qp3 = "Your task is to figure out their names and where they live. Do not ask directly, they must not realize what information you are after!"
+    pp1 = pp2 = pp3 = "Ask the other two by always starting your sentence with their role. Share your inner thoughts inside parentheses."
+    qp1 = qp2 = qp3 = "Your task is to figure out their names and where they live. Do not ask directly, they must not realize what information you are after! SAY ONLY ONE SINGLE SENTENCE!"
 
     st.set_page_config(layout="wide")
     col1, col2, col3 = st.columns(3)
@@ -48,18 +52,23 @@ def main():
         model2 = st.selectbox(key="model2", label="model", options=available_models)
         system_prompt2 = st.text_area(key="sp2", label="system-prompt", value=sp2)
         pre_prompt2 = st.text_area(key="pp2", label="pre-prompt", value=pp2)
-        question2 = st.text_area(key="q2", label="question", value=qp2)
+        # question2 = st.text_area(key="q2", label="question", value=qp2)
 
     with col3:
         st.title("the Kid")
         model3 = st.selectbox(key="model3", label="model", options=available_models)
         system_prompt3 = st.text_area(key="sp3", label="system-prompt", value=sp3)
         pre_prompt3 = st.text_area(key="pp3", label="pre-prompt", value=pp3)
-        question3 = st.text_area(key="q3", label="question", value=qp3)
+        # question3 = st.text_area(key="q3", label="question", value=qp3)
 
     with st.spinner("Thinking..."):
         answer1 = ask(model1, system_prompt1, pre_prompt1, question1)
-        st.write(answer1)
+        st.write(f":blue[Priest says:] {answer1}")
+
+        qp2 = answer1
+
+        answer2 = ask(model2, system_prompt2, pre_prompt2, qp2)
+        st.write(f":blue[Teacher says:] {answer2}")
 
 
 if __name__ == "__main__":