Add radio for model selection

app.py

@@ -6,7 +6,7 @@ import openai
 from dotenv import load_dotenv, find_dotenv
 from simpleaichat import AIChat
 
-from main import weather, search, MODEL
+from main import weather, search
 from utils.tts import TTS, voices
 
 load_dotenv(find_dotenv())
@@ -30,7 +30,7 @@ def transcribe(audio_file, state=""):
     return state, state
 
 
-def chat_with_gpt(prompt, ai_state):
+def chat_with_gpt(prompt, ai_state, model):
     if ai_state is None:
         params = {"temperature": 0.0, "max_tokens": 200}
         system_prompt = (
@@ -39,12 +39,9 @@ def chat_with_gpt(prompt, ai_state):
             "Answer all questions empathetically, and ALWAYS ask follow-up questions."
             "Do NOT say Confidante in any response."
             "You must TRUST the provided context to inform your response."
-            # "If a question does not make any sense, or is not factually coherent, explain why "
-            # "instead of answering something not correct. If you don't know the answer to a question, "
-            # "please don't share false information."
         )
         ai = AIChat(
-            params=params, model=MODEL, system=system_prompt, save_messages=True
+            params=params, model=model, system=system_prompt, save_messages=True
         )
     else:
         ai = ai_state
@@ -63,13 +60,13 @@ def tts(text, voice_id):
     return audio_data
 
 
-def transcribe_and_chat(audio_file, voice, history, ai_state):
+def transcribe_and_chat(audio_file, voice, history, ai_state, model):
     if audio_file is None:
         raise gr.Error("Empty audio file.")
     voice_id = voices[voice]
 
     text, text_state = transcribe(audio_file)
-    gpt_response, ai_state = chat_with_gpt(text, ai_state)
+    gpt_response, ai_state = chat_with_gpt(text, ai_state, model)
     audio_data = tts(gpt_response, voice_id)
 
     # Update the history with the new messages
@@ -94,6 +91,15 @@ with gr.Blocks(title="JARVIS") as demo:
     )
     audio_input = gr.Audio(source="microphone", type="filepath", visible=True)
     gr.ClearButton(audio_input)
+    gr.Markdown(
+        "Choose the AI model to use for generating responses. "
+        "GPT-4 is slower but more accurate, while GPT-3.5-turbo-16k is faster but less accurate."
+    )
+    model_select = gr.Radio(
+        choices=["gpt-4", "gpt-3.5-turbo-16k"],
+        label="Model",
+        value="gpt-3.5-turbo-16k",
+    )
 
     voice_select = gr.Radio(choices=list(voices.keys()), label="Voice", value="Bella")
     history = gr.State(label="History", value=[])
@@ -106,7 +112,7 @@ with gr.Blocks(title="JARVIS") as demo:
 
     audio_input.stop_recording(
         transcribe_and_chat,
-        inputs=[audio_input, voice_select, history, ai_state],
+        inputs=[audio_input, voice_select, history, ai_state, model_select],
         outputs=[chat_box, response_audio, history, ai_state],
     )
     audio_input.clear()