process_qid_to_data

app.py

@@ -247,7 +247,10 @@ def create_csv(processed_data):
     return file_path
 
 def process_single_image(image):
-    if isinstance(image, str):  # If the image is a file path
+    if isinstance(image, str) and image.startswith("http"):
+        # If the image is a URL, use it directly
+        image_url = image
+    elif isinstance(image, str):  # If the image is a file path
         image_url = upload_image_to_gcs(image, bucket)
     else:  # If the image is an image object (from gr.Image)
         temp_file_path = "/tmp/temp_image.png"
@@ -493,6 +496,31 @@ def show_multiple_questions_markdown(data):
 
     return markdown
 
+
+# ====== Junyi Q_ID ======
+def process_qid_to_data(password, q_id):
+    # 檢查密碼
+    if password != PASSWORD:
+        raise gr.Error("密码错误，请重新输入")
+    # 根據 Junyi Q_ID 取得題目截圖
+    processed_data = []
+    # 獲取圖片的方法。格式：https://storage.googleapis.com/exercise-render-img-junyi/P-qid.jpg
+    image_url = f"https://storage.googleapis.com/exercise-render-img-junyi/p_{q_id}.jpg"
+    # 處理圖片
+    text, question_json = process_image(image_url)
+    # 生成 Perseus JSON
+    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)
+    
+    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:
     with gr.Row():
@@ -556,6 +584,26 @@ with gr.Blocks() as demo:
                     wrap=True
                 )
 
+    with gr.Tab("Junyi_Q_ID"):
+        with gr.Row():
+            gr.Markdown("## Junyi Q_ID")
+        with gr.Row():
+            junyi_q_id_input = gr.Textbox(label="Junyi Q_ID")
+            junyi_q_id_submit_button = gr.Button("開始處理 Junyi Q_ID")
+        with gr.Row():
+            junyi_q_id_result_text = gr.Textbox(label="處理結果")
+            junyi_q_id_download_csv_output = gr.File(label="下载 CSV")
+        with gr.Row():
+            junyi_q_id_question_image = gr.Image()
+            junyi_q_id_question_markdown = gr.Markdown(show_label=False, latex_delimiters=[{"left": "$", "right": "$", "display": False}])
+        with gr.Accordion(open=False):
+            with gr.Row():
+                junyi_q_id_result_table = gr.Dataframe(
+                    headers=["圖片URL", "文字", "題號", "題目", "選項1", "選項2", "選項3", "選項4", "答案", "提示1", "提示2", "提示3", "提示4", "提示5", "Perseus JSON"],
+                    column_widths=[10, 10, 5, 20, 4, 4, 4, 4, 4,4,4,4,4,4, 10],
+                    wrap=True
+                )
+        
     submit_button.click(
         fn=process_image_to_data,
         inputs=[password_input, image_input],
@@ -590,6 +638,20 @@ with gr.Blocks() as demo:
         outputs=[pdf_question_markdown]
     )
 
+    junyi_q_id_submit_button.click(
+        fn=process_qid_to_data,
+        inputs=[password_input, junyi_q_id_input],
+        outputs=[junyi_q_id_result_table, junyi_q_id_result_text, junyi_q_id_download_csv_output]
+    ).then(
+        fn=show_single_question_image,
+        inputs=[junyi_q_id_result_table],
+        outputs=[junyi_q_id_question_image]
+    ).then(
+        fn=show_single_question_markdown,
+        inputs=[junyi_q_id_result_table],
+        outputs=[junyi_q_id_question_markdown]
+    )
+
 demo.launch()