Update app.py

app.py

@@ -3,8 +3,6 @@ from openai import OpenAI
 from os import getenv
 import json
 
-MODEL_ID = "mistralai/mixtral-8x22b-instruct"
-
 client = OpenAI(
   base_url="https://openrouter.ai/api/v1",
   api_key=getenv("OPENROUTER_API_KEY"),
@@ -14,9 +12,9 @@ client = OpenAI(
 with gr.Blocks() as app:
     dialog_state_var = gr.State([])
     
-    def api_call(prompt):
+    def api_call(prompt, model_id):
         completion = client.chat.completions.create(
-        model=MODEL_ID,
+        model=model_id,
         messages=[
             {
                 "role": "user",
@@ -27,7 +25,7 @@ with gr.Blocks() as app:
         answer = completion.choices[0].message.content
         return answer
     
-    def generate_next(intro, setting, topic, student_identity, teacher_identity, student_start, student_base, teacher_base, dialog_state):
+    def generate_next(intro, setting, topic, student_identity, teacher_identity, student_start, student_base, teacher_base, dialog_state, llm_teacher, llm_student):
     
         dialog_prev = "\n\n".join(dialog_state)
     
@@ -37,7 +35,7 @@ with gr.Blocks() as app:
             prompt_student+=student_start
         else:
             prompt_student+=student_base
-        student = api_call(prompt_student)
+        student = api_call(prompt_student, llm_student)
         print(student)
         student = json.loads(student)
         analysis_student = student["analyse"]
@@ -47,7 +45,7 @@ with gr.Blocks() as app:
     
         # TEACHER
         prompt_teacher = intro+"\n\Verlauf des Dialogs: "+setting+"\n\nThema, bei dem der Schüler Hilfe braucht:"+topic+"\n\nBeschreibung des Schülers:"+student_identity+"\n\nBeschreibung des Tutors:"+teacher_identity+"\n\nBisheriger Dialog:\n"+dialog_prev+"\n\n"+teacher_base
-        teacher = api_call(prompt_teacher)
+        teacher = api_call(prompt_teacher, llm_teacher)
         print(teacher)
         teacher = json.loads(teacher)
         analysis_teacher = teacher["analyse"]
@@ -67,6 +65,8 @@ with gr.Blocks() as app:
 
     
     # Text boxes for user inputs
+    txt_llm_teacher = gr.Textbox(label="LLM Tutor (https://openrouter.ai/models)", lines=4, value = "mistralai/mixtral-8x22b-instruct")
+    txt_llm_student = gr.Textbox(label="LLM Schüler (https://openrouter.ai/models)", lines=4, value = "mistralai/mixtral-8x22b-instruct")
     txt_intro = gr.Textbox(label="Einleitung", lines=4, value = "Ich möchte einen möglichst realistischen Dialog zwischen einem Schüler und einem Tutor simulieren.") 
     txt_setting = gr.Textbox(label="Dialogverlauf", lines=4, value = "Der Schüler kommt bei seinen Hausaufgaben nicht weiter und bittet den Tutorum Hilfe. Der Tutor diagnostiziert welches Vorwissen der Schülers hat und welche Art von Hilfe er genau braucht. Dann erabeiten Tutor und Schüler Schritt für Schritt das Thema.")
     txt_topic = gr.Textbox(label="Thema des Dialogs", lines=4, value="Zellteilung - Die Phasen der Mitose")
@@ -89,7 +89,7 @@ with gr.Blocks() as app:
     # Define the layout and how components interact
     btn_submit.click(
         fn=generate_next,
-        inputs=[txt_intro, txt_setting, txt_topic, txt_student_identity, txt_teacher_identity,txt_student_start, txt_student_base, txt_teacher_base, dialog_state_var],
+        inputs=[txt_intro, txt_setting, txt_topic, txt_student_identity, txt_teacher_identity,txt_student_start, txt_student_base, txt_teacher_base, dialog_state_var,txt_llm_teacher,txt_llm_student],
         outputs=[prompt_student,prompt_teacher, chat_interface, dialog_state_var, analysis_student, analysis_teacher],
     )