Fix: Add provider parameter to rewrite_message function to fix NameError

app.py

@@ -13,7 +13,7 @@ os.environ["STREAMLIT_SERVER_ADDRESS"] = "0.0.0.0"
 
 
 def rewrite_message(
-    msg: str, persona_name: str, show_rationale: bool, skip_rewrite: bool
+    msg: str, persona_name: str, show_rationale: bool, skip_rewrite: bool, provider: str = "openai"
 ) -> str:
     if skip_rewrite:
         rewritten_msg = msg
@@ -140,13 +140,13 @@ if msg:
     # rewritten_msg = "Use the persona profile to personalize your naswer only when applicable.\n"
     if compare_personas:
         all_answers = {}
-        rewritten_msg = rewrite_message(msg, persona_name, show_rationale, False)
+        rewritten_msg = rewrite_message(msg, persona_name, show_rationale, False, provider)
         msgs = clean_history(st.session_state.history, persona_name)
         msgs = append_user_turn(msgs, rewritten_msg)
         txt, lat, tok, tps = chat(msgs, persona_name)
         all_answers[persona_name] = txt
 
-        rewritten_msg_control = rewrite_message(msg, "Control", show_rationale, True)
+        rewritten_msg_control = rewrite_message(msg, "Control", show_rationale, True, provider)
         msgs_control = clean_history(st.session_state.history, "Control")
         msgs_control = append_user_turn(msgs_control, rewritten_msg_control)
         txt_control, lat, tok, tps = chat(msgs_control, "Arnold")
@@ -156,7 +156,7 @@ if msg:
             {"role": "assistant_all", "axis": "role", "content": all_answers}
         )
     else:
-        rewritten_msg = rewrite_message(msg, persona_name, show_rationale, skip_rewrite)
+        rewritten_msg = rewrite_message(msg, persona_name, show_rationale, skip_rewrite, provider)
         msgs = clean_history(st.session_state.history, persona_name)
         msgs = append_user_turn(msgs, rewritten_msg)
         txt, lat, tok, tps = chat(