main.py

import json
import time

import requests

# 讀取主辦提供的 Question JSON 檔案
with open('data/questions.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} 秒')

total_end_time = time.time()
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)

print('合併輸出已保存到 pred_retrieve.json 文件中。')