Update app.py

app.py

@@ -10,52 +10,53 @@ client = OpenAI(
 )
 
 # Initialize the chat log
-dialog = []
-dialog_state = gr.State([])
-
-def api_call(prompt):
-    completion = client.chat.completions.create(
-    model=MODEL_ID,
-    messages=[
-        {
-            "role": "user",
-            "content": prompt,
-        },
-    ],
-    )
-    answer = completion.choices[0].message.content
-    return answer
-
-def generate_next(intro, setting, topic, student_identity, teacher_identity, student_base, teacher_base):
-
-    dialog_prev = "\n\n".join(dialog_state)
-
-    # STUDENT
-    student_start = "Basierend auf diesen Vorgaben, überlege, wie der Schüler den Dialog starten würde und schreibe einen realistischen ersten Satz des Schülers. Nutze folgendes Format S: <Schüler Antwort>"
-    prompt_student = 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"
-    if len(dialog) == 0:
-        prompt_student+=student_start
-    else:
-        prompt_student+=student_base
-    student = api_call(prompt_student)
-    dialog_state.append(student)
-    dialog_prev = "\n\n".join(dialog_state)
-
-    # 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)
-    dialog_state.append(teacher)
+with gr.Blocks() as app:
+    dialog = []
+    dialog_state_var = gr.State([])
     
-    dialog_new = "\n\n".join(dialog_state)
+    def api_call(prompt):
+        completion = client.chat.completions.create(
+        model=MODEL_ID,
+        messages=[
+            {
+                "role": "user",
+                "content": prompt,
+            },
+        ],
+        )
+        answer = completion.choices[0].message.content
+        return answer
     
-    return prompt_student,prompt_teacher,dialog_new
-
-
-def reset():
-    dialog = []
+    def generate_next(intro, setting, topic, student_identity, teacher_identity, student_base, teacher_base, dialog_state):
+    
+        dialog_prev = "\n\n".join(dialog_state)
+    
+        # STUDENT
+        student_start = "Basierend auf diesen Vorgaben, überlege, wie der Schüler den Dialog starten würde und schreibe einen realistischen ersten Satz des Schülers. Nutze folgendes Format S: <Schüler Antwort>"
+        prompt_student = 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"
+        if len(dialog) == 0:
+            prompt_student+=student_start
+        else:
+            prompt_student+=student_base
+        student = api_call(prompt_student)
+        dialog_state.append(student)
+        dialog_prev = "\n\n".join(dialog_state)
     
+        # 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)
+        dialog_state.append(teacher)
+        
+        dialog_new = "\n\n".join(dialog_state)
+        
+        return prompt_student,prompt_teacher, dialog_new, dialog_state
+    
+    
+    def reset():
+        dialog = []
+        
+
 
-with gr.Blocks() as app:
     
     # Text boxes for user inputs
     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.") 
@@ -77,8 +78,8 @@ 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_base, txt_teacher_base],
-        outputs=[prompt_student,prompt_teacher, chat_interface],
+        inputs=[txt_intro, txt_setting, txt_topic, txt_student_identity, txt_teacher_identity, txt_student_base, txt_teacher_base, dialog_state_var],
+        outputs=[prompt_student,prompt_teacher, chat_interface, dialog_state_var],
     )