Update pages/2_phonics.py

pages/2_phonics.py

@@ -164,11 +164,19 @@ def process_response(user_input: str, conversation_history: list) -> tuple:
  model_response, conversation_history = generate_response(initial_prompt, conversation_history)
  else:
  model_response, conversation_history = generate_response(user_input, conversation_history)
- lessonList = ast.literal_eval(model_response)
+
+ pattern = re.compile(r"\[(.*?)\]", re.DOTALL)
+
+ # Find the match
+ match = pattern.search(model_response)
+ 
+ list_content = f"[{match.group(1)}]"
+
+ lessonList = ast.literal_eval(list_content)
  return lessonList, conversation_history
  
 @st.cache_data
-def generate_image(prompt: str) -> str:
+def get_image(prompt: str) -> str:
  try:
  return findImg(prompt)
  except:
@@ -405,7 +413,7 @@ def main():
  
  with col2:
  if 'img_prompt' in st.session_state:
- st.session_state['img_path']=generate_image(st.session_state['img_prompt'])
+ st.session_state['img_path']=get_image(st.session_state['img_prompt'])
  del st.session_state['img_prompt']
  
  st.image(st.session_state['img_path'], caption="Generated Image")