Update app.py

app.py

@@ -24,6 +24,18 @@ with gr.Blocks() as app:
         )
         answer = completion.choices[0].message.content
         return answer
+
+    def format(str):
+        s = str.split("\",")
+        for i in range(len(s)):
+            key = s[i].split(":")[0]
+            value = s[i].split(":")[1]
+            value = value.replace("\"", "")
+            s[i]=key+":"+"\""+value+"\""
+        str_new = ",".join(s)
+        str_new = str_new.replace("}\"", "\"}")
+        str_new = str_new.replace("\n", "")
+        return str_new
     
     def generate_next(intro, setting, topic, student_identity, teacher_identity, student_start, student_base, teacher_base, dialog_state, llm_teacher, llm_student):
     
@@ -37,6 +49,7 @@ with gr.Blocks() as app:
             prompt_student+=student_base
         student = api_call(prompt_student, llm_student)
         print(student)
+        student = format(student)
         student = json.loads(student,strict=False)
         analysis_student = student["analyse"]
         response = "Schüler: "+student["antwort"]
@@ -46,6 +59,7 @@ with gr.Blocks() as app:
         # TEACHER
         prompt_teacher = intro+"\n\nBisheriger Dialog:\n"+dialog_prev+"\n\Verlauf des Dialogs: "+setting+"\n\nThema, bei dem der Schüler Hilfe braucht:"+topic+"\n\nBeschreibung des Tutors:"+teacher_identity+"\n\n"+teacher_base
         teacher = api_call(prompt_teacher, llm_teacher)
+        teacher = format(teacher)
         print(teacher)
         teacher = json.loads(teacher,strict=False)
         analysis_teacher = teacher["analyse"]