summarise_json

app.py

@@ -438,13 +438,30 @@ def generate_summarise(df_string):
         以及可能的結論與結尾延伸小問題提供學生作反思
 
         整體格式為：
-        1. 內容類型：？
+        1. 內容類型
         2. 整體摘要
         3. 條列式重點
         4. 關鍵時刻（段落摘要）
         5. 結論反思（為什麼我們要學這個？）
         6. 延伸小問題
 
+        JSON format 為 {
+            "content_type": "資料",
+            "summary": "整體摘要",
+            "key_points": "條列式重點",
+            "key_moments": [
+                {
+                    "start": "開始時間",
+                    "end": "結束時間",
+                    "text": "段落摘要"
+                },
+                ...
+            ],
+            ]",
+            "conclusion_reflection": "結論反思（為什麼我們要學這個？）",
+            "extension_questions": "延伸小問題 LIST"
+        }
+
     """
 
     # 🗂️ 1. 內容類型：？
@@ -463,15 +480,20 @@ def generate_summarise(df_string):
         "model": "gpt-4-1106-preview",
         "messages": messages,
         "max_tokens": 4000,
+        "response-format": {"type": "json_object"}
     }
 
     response = client.chat.completions.create(**request_payload)
-    df_summarise = response.choices[0].message.content.strip()
-    print("=====df_summarise=====")
-    print(df_summarise)
-    print("=====df_summarise=====")
+    print("====response====")
+    print(response)
+    print("====response====")
+
+    summarise_json = response.choices[0].message.content
+    print("=====summarise_json=====")
+    print(summarise_json)
+    print("=====summarise_json=====")
 
-    return df_summarise
+    return summarise_json
 
 
 def generate_questions(df_string):