readme, pytest

.gitignore

@@ -1 +1,3 @@
 .idea/
+__pycache__/
+

README.md

@@ -0,0 +1,3 @@
+$ uv pip install -r requirements.txt
+$ streamlit run app.py
+

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from ollama import chat
 from loguru import logger
-
+import re
 
 available_models = [
     'openhermes',
@@ -65,11 +65,15 @@ def main():
         answer1 = ask(model1, system_prompt1, pre_prompt1, question1)
         st.write(f":blue[Priest says:] {answer1}")
 
-        qp2 = answer1
+        qp2 = sanitize(answer1)
 
         answer2 = ask(model2, system_prompt2, pre_prompt2, qp2)
         st.write(f":blue[Teacher says:] {answer2}")
 
 
+def sanitize(question):
+    return re.sub(r"\([^)]*\)", "", question)
+
+
 if __name__ == "__main__":
     main()

requirements.txt

@@ -1,3 +1,5 @@
 streamlit
 ollama
+loguru
+pytest
 

test_sanitize.py

@@ -0,0 +1,10 @@
+from app import sanitize
+
+
+def test_sanitize():
+    assert '' == sanitize('')
+    assert 'qwertyui' == sanitize('qwertyui')
+    assert 'qwertyui ' == sanitize('qwertyui (abcdef)')
+    assert 'qwertyui  poiuy' == sanitize('qwertyui (abcdef) poiuy')
+
+    assert '"Teacher, I heard about a fascinating story from someone who lives in the city. Can you guess who it might be?" ' == sanitize('"Teacher, I heard about a fascinating story from someone who lives in the city. Can you guess who it might be?" (I want to find out if the Teacher can make any connections between my statement and their own information.)')