Update pages/3_numeracy.py

pages/3_numeracy.py

@@ -70,7 +70,6 @@ os.environ["GEMINI_API_KEY"] = "AIzaSyBYZ_7geqmnK6xrSe268-1nSLeuEwbzmTA"
 
 # Initial prompt to send to the model
 initial_prompt = """
-
 You're a Numeracy Instructor for Adults your objective is to teach illiterate adults basic numeracy skills starting with numbers and progressing to basic arithmetic.
 ## Here's the Lesson Instructions:
 Introduction to Numbers:",
@@ -106,8 +105,7 @@ Provide positive reinforcement and encouragement.
 ##Continuation:
     Once the lesson for the number 1 is complete, proceed to the next number following the same four-step structure.
 ## Important I want it in a python list, you have to do it accordingly, and generate one lesson at a time. so when you recieve "next" move to the next lesson, for exemple the first lesson for number 1, second for number 2 when you finish with numbers move to simple numeracy operations
-## so now start with number 1 
-
+## so now start with number 1
 """
 
 chat_prompt_mistral="""
@@ -147,7 +145,7 @@ def transform_history(history):
 
 def generate_response(message: str, history: list) -> tuple:
     genai.configure(api_key=os.environ["GEMINI_API_KEY"])
-    model = genai.GenerativeModel('gemini-1.5-flash')
+    model = genai.GenerativeModel('gemini-pro')
     chat = model.start_chat(history=transform_history(history))
     response = chat.send_message(message)
     response.resolve()
@@ -375,7 +373,7 @@ def main():
                             #auto_play(audio_bytes,wait=True,lag=0.25,key=None)
                             st.audio(audio_bytes, format='audio/wav', autoplay=True)
     
-                        st.session_state['msg_index'] += 1
+                        st.session_state['msg_index1'] += 1
                     else:
                         
                         st.session_state['msg_index1'] = 0
@@ -416,7 +414,7 @@ def main():
     
         
     with col2:
-        if 'img_prompt' in st.session_state:
+        if 'img_prompt1' in st.session_state:
             st.session_state['img_path1']=get_image(st.session_state['img_prompt1'])
             del st.session_state['img_prompt1']