create_csv

app.py

@@ -8,6 +8,8 @@ from google.cloud import storage
 from google.oauth2 import service_account
 from googleapiclient.http import MediaIoBaseDownload
 from storage_service import GoogleCloudStorage
+import csv
+import io
 
 
 is_env_local = os.getenv("IS_ENV_LOCAL", "false") == "true"
@@ -37,6 +39,8 @@ bucket = GCS_CLIENT.bucket(bucket_name)
 GSHEET_KEY_DICT = json.loads(GSHEET_KEY)
 sheets_client = gspread.service_account_from_dict(GSHEET_KEY_DICT)
 
+CSV_DATA = []
+
 # 函数定义
 def upload_image_to_gcs(image_path, bucket):
     # Extract the file name from the path and create a unique filename
@@ -125,58 +129,68 @@ def text_to_json(text):
     result = response_to_json.choices[0].message.content
     return result
 
-def upload_to_sheet(sheet, image_url, text, question_json):
-    if image_url and text and question_json:
-        q_id = question_json['q_id']
-        question = question_json['question']
-        choice_1 = question_json['choice_1']
-        choice_2 = question_json['choice_2']
-        choice_3 = question_json['choice_3']
-        choice_4 = question_json['choice_4']
-        perseus_text = """{"correct_nxt_qid": null, "wrong_nxt_qid": null, "itemDataVersion": {"major": 0, "minor": 1}, "question": {"content": "", "images": {}, "widgets": {"radio 1": {"version": {"major": 0, "minor": 0}, "type": "radio", "graded": true, "options": {"onePerLine": true, "noneOfTheAbove": false, "choices": [{"content": "", "correct": false}, {"content": "", "correct": false}, {"content": "", "correct": false}, {"content": "", "correct": false}], "displayCount": null, "multipleSelect": false, "randomize": false}}}}, "answerArea": {"calculator": false, "type": "multiple", "options": {}}, "is_start": false, "hints": [{"content": "", "images": {}, "widgets": {}}, {"content": "", "images": {}, "widgets": {}}]}"""
-        perseus_json = json.loads(perseus_text)
-        widget = "\n\n[[☃ radio 1]]"
-        perseus_json["question"]["content"] = question + widget
-        perseus_json["question"]["widgets"]["radio 1"]["options"]["choices"][0]["content"] = "$\\mbox{(A)}$ " + choice_1
-        perseus_json["question"]["widgets"]["radio 1"]["options"]["choices"][1]["content"] = "$\\mbox{(B)}$ " + choice_2
-        perseus_json["question"]["widgets"]["radio 1"]["options"]["choices"][2]["content"] = "$\\mbox{(C)}$ " + choice_3
-        perseus_json["question"]["widgets"]["radio 1"]["options"]["choices"][3]["content"] = "$\\mbox{(D)}$ " + choice_4
-        perseus_json_str = json.dumps(perseus_json)
-        sheet.append_row([image_url, text, q_id, question, choice_1, choice_2, choice_3, choice_4, perseus_json_str])
-    else:
-        print("圖片處理失敗，未上傳到Google Sheets")
-
-def process_and_upload(images, sheet_url):
-    verify_password(password_input)
+def build_perseus_json(question_json):
+    question = question_json['question']
+    choice_1 = question_json['choice_1']
+    choice_2 = question_json['choice_2']
+    choice_3 = question_json['choice_3']
+    choice_4 = question_json['choice_4']
+    perseus_text = """{"correct_nxt_qid": null, "wrong_nxt_qid": null, "itemDataVersion": {"major": 0, "minor": 1}, "question": {"content": "", "images": {}, "widgets": {"radio 1": {"version": {"major": 0, "minor": 0}, "type": "radio", "graded": true, "options": {"onePerLine": true, "noneOfTheAbove": false, "choices": [{"content": "", "correct": false}, {"content": "", "correct": false}, {"content": "", "correct": false}, {"content": "", "correct": false}], "displayCount": null, "multipleSelect": false, "randomize": false}}}}, "answerArea": {"calculator": false, "type": "multiple", "options": {}}, "is_start": false, "hints": [{"content": "", "images": {}, "widgets": {}}, {"content": "", "images": {}, "widgets": {}}]}"""
+    perseus_json = json.loads(perseus_text)
+    widget = "\n\n[[☃ radio 1]]"
+    perseus_json["question"]["content"] = question + widget
+    perseus_json["question"]["widgets"]["radio 1"]["options"]["choices"][0]["content"] = "$\\mbox{(A)}$ " + choice_1
+    perseus_json["question"]["widgets"]["radio 1"]["options"]["choices"][1]["content"] = "$\\mbox{(B)}$ " + choice_2
+    perseus_json["question"]["widgets"]["radio 1"]["options"]["choices"][2]["content"] = "$\\mbox{(C)}$ " + choice_3
+    perseus_json["question"]["widgets"]["radio 1"]["options"]["choices"][3]["content"] = "$\\mbox{(D)}$ " + choice_4
+    perseus_json_str = json.dumps(perseus_json)
+    return perseus_json_str
+
+def create_csv(processed_data):
+    # 设定一个可写的目录路径
+    writable_directory = "/tmp/csv_files"
+    if not os.path.exists(writable_directory):
+        os.makedirs(writable_directory)  # 如果目录不存在，创建它
+
+    timestamp = int(time.time())
+    file_name = f"csv_{timestamp}.csv"
+    file_path = os.path.join(writable_directory, file_name)
+
+    # 创建并写入 CSV 文件
+    with open(file_path, mode='w', newline='', encoding='utf-8') as file:
+        writer = csv.writer(file)
+
+        # 写入标题行
+        headers = ["圖片URL", "文字", "題號", "題目", "選項1", "選���2", "選項3", "選項4", "Perseus JSON"]
+        writer.writerow(headers)
+
+        # 写入数据行
+        for row in processed_data:
+            writer.writerow(row)
+
+    return file_path
+
+def process_image_to_data(password, images):
+    if password != PASSWORD:
+        raise gr.Error("密码错误，请重新输入")
     
-    sheet = sheets_client.open_by_url(sheet_url).sheet1
-    all_data = sheet.get_all_values()
-    sheet_start = len(all_data) + 1  # 現有數據的下一行
+    processed_data = []
     for image_file in images:
         image_url = upload_image_to_gcs(image_file, bucket)
-        text, question_json = process_image(image_url)
-        upload_to_sheet(sheet, image_url, text, question_json)
+        text, question_json = process_image(image_url)        
+        perseus_json_str = build_perseus_json(question_json)
+
+        # 添加數據到列表，用於後續顯示
+        processed_data.append([image_url, text] + list(question_json.values()) + [perseus_json_str])
+
         print("======process_and_upload=====")
         print("image_url:", image_url)
-        
-    all_data = sheet.get_all_values()
-    sheet_end = len(all_data)  # 最後一個有數據的行
-    question_count = sheet_end - sheet_start + 1
-    result = f"""
-    圖片處理完成，已上傳到Google Sheets
-    sheet_url: {sheet_url}
-    起始列: {sheet_start}
-    結束列: {sheet_end} 
-    總共完成 {question_count} 道題目
-    """
-
-    return result
 
-def verify_password(password):
-    if password == PASSWORD:
-        return True
-    else:
-        return gr.Error("密码错误")
+    question_count = len(processed_data)
+    result = f"圖片處理完成，總共完成 {question_count} 道題目"
+    csv_file_path = create_csv(processed_data)
+    
+    return processed_data, result, csv_file_path
 
 # Gradio界面
 with gr.Blocks() as demo:
@@ -185,14 +199,27 @@ with gr.Blocks() as demo:
         password_input = gr.Textbox(label="密码", type="password")
     with gr.Row():
         image_input = gr.Files(label="选择图片", type="filepath")
-        sheet_default_value = "https://docs.google.com/spreadsheets/d/1ygFGLxcnPad3LMVj4bZqfGZh1n2wqhs0-vOUjuVCkSY/edit#gid=0"
-        sheet_input = gr.Textbox(label="Google Sheets URL", value=sheet_default_value)
+        # sheet_default_value = "https://docs.google.com/spreadsheets/d/1ygFGLxcnPad3LMVj4bZqfGZh1n2wqhs0-vOUjuVCkSY/edit#gid=0"
+        # sheet_input = gr.Textbox(label="Google Sheets URL", value=sheet_default_value)
         submit_button = gr.Button("開始處理圖片")
+    with gr.Row():
+        result_text = gr.Textbox(label="處理結果")
+    with gr.Row():
+        result_table = gr.Dataframe(
+            headers=["圖片URL", "文字", "題號", "題目", "選項1", "選項2", "選項3", "選項4", "Perseus JSON"],
+            column_widths=[10, 25, 5, 20, 5, 5, 5, 5, 20],
+            wrap=True
+        )
+
+        
+    with gr.Row():
+        download_csv_output = gr.File(label="下载 CSV")
+        
 
     submit_button.click(
-        fn=process_and_upload,
-        inputs=[image_input, sheet_input],
-        outputs=gr.Markdown()
+        fn=process_image_to_data,
+        inputs=[password_input, image_input],
+        outputs=[result_table, result_text, download_csv_output]
     )
 
 demo.launch()