Add comments for main.py

main.py

@@ -3,31 +3,40 @@
 
 import requests
 
+# 讀取主辦提供的 Question JSON 檔案
 with open('data/questions_example.json', encoding='utf-8') as file:
     questions = json.load(file)['questions']
 
+# 初始化輸出資料格式
 output_data = {'answers': []}
 
+# 設定 Flask 應用程式的 URL
 url = 'http://127.0.0.1:5000/api/chat'
 
+# 計算總花費時間
 total_start_time = time.time()
 
 for question in questions:
+    # 計算每個問題的處理時間
     question_start_time = time.time()
 
+    # 發送 POST 請求到 Model/flask_app.py 的 Retrieve API 端點
     response = requests.post(url, json=question)
 
     if response.status_code == 200:
         response_json = response.json()
 
+        # 從回應中提取 qid 和 retrieve 欄位
         qid = question.get('qid')
         retrieve = response_json.get('retrieve')
 
+        # 將 qid 和 retrieve 加入輸出資料中
         output_data['answers'].append({'qid': qid, 'retrieve': retrieve})
         print('成功取得 JSON:', response_json)
     else:
         print('請求失敗，狀態碼:', response.status_code)
 
+    # 計算每個問題的處理時間
     question_end_time = time.time()
     question_duration = question_end_time - question_start_time
     print(f'QID: {qid} - 花費時間: {question_duration:.2f} 秒')
@@ -36,6 +45,7 @@
 total_duration = total_end_time - total_start_time
 print(f'全部題目處理完成，總共花費時間: {total_duration:.2f} 秒')
 
+# 將輸出資料寫入 JSON 文件
 with open('data/pred_retrieve.json', 'w', encoding='utf-8') as output_file:
     json.dump(output_data, output_file, ensure_ascii=False, indent=4)