with gr.Tab("單張處理"):

app.py

@@ -242,6 +242,10 @@ def process_image_to_data(password, images):
         raise gr.Error("密码错误，请重新输入")
     
     processed_data = []
+
+    if isinstance(images, str):
+        images = [images]
+        
     for image_file in images:
         image_url = upload_image_to_gcs(image_file, bucket)
         text, question_json = process_image(image_url)        
@@ -259,28 +263,94 @@ def process_image_to_data(password, images):
     
     return processed_data, result, csv_file_path
 
+def show_single_question_image(data):
+    if len(data) == 0:
+        return ""
+    question_json = data.iloc[0].to_dict()  # 確保訪問的是 DataFrame 的第一行並轉換為字典
+    image_url = question_json['圖片URL']
+    return image_url
+
+
+def show_single_question_markdown(data):
+    if len(data) == 0:
+        return ""
+    question_json = data.iloc[0].to_dict()  # 確保訪問的是 DataFrame 的第一行並轉換為字典
+    question = question_json['題目']
+    choice_1 = question_json['選項1']
+    choice_2 = question_json['選項2']
+    choice_3 = question_json['選項3']
+    choice_4 = question_json['選項4']
+    answer = question_json['答案']
+    
+    hints = []
+    for i in range(1, 6):
+        hint_key = question_json.get(f'提示{i}', None)
+        if hint_key:
+            hints.append(hint_key)
+        else:
+            break
+
+
+    markdown = f"""
+## 題目
+- {question}
+
+## 選項
+1. {choice_1}
+2. {choice_2}
+3. {choice_3}
+4. {choice_4}
+
+## 答案: {answer}
+
+## 提示
+"""
+    for i, hint in enumerate(hints):
+        markdown += f"{i+1}. {hint}\n"
+
+    return markdown
+
 # Gradio界面
 with gr.Blocks() as demo:
-    gr.Markdown("## 批量上传图片并处理")
     with gr.Row():
         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)
-        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", "答案", "提示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
-        )
-
-        
-    with gr.Row():
-        download_csv_output = gr.File(label="下载 CSV")
+    
+    with gr.Tab("批量處理"):
+        with gr.Row():
+            gr.Markdown("## 批量图片處理 + Perseus JSON 生成")
+            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)
+            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", "答案", "提示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
+            )
+        with gr.Row():
+            download_csv_output = gr.File(label="下载 CSV")
+    
+    with gr.Tab("單張處理"):
+        with gr.Row():
+            gr.Markdown("## 單張图片處理")
+            single_image_input = gr.Files(label="選擇圖片", type="filepath")
+            single_submit_button = gr.Button("開始處理圖片")
+        with gr.Row():
+            single_result_text = gr.Textbox(label="處理結果")
+        with gr.Row():
+            single_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
+            )
+        with gr.Row():
+            single_question_image = gr.Image()
+            single_question_markdown = gr.Markdown(show_label=False, latex_delimiters=[{"left": "$", "right": "$", "display": False}]) 
+        with gr.Row():
+            single_download_csv_output = gr.File(label="下载 CSV")
         
 
     submit_button.click(
@@ -289,4 +359,18 @@ with gr.Blocks() as demo:
         outputs=[result_table, result_text, download_csv_output]
     )
 
+    single_submit_button.click(
+        fn=process_image_to_data,
+        inputs=[password_input, single_image_input],
+        outputs=[single_result_table, single_result_text, single_download_csv_output]
+    ).then(
+        fn=show_single_question_image,
+        inputs=[single_result_table],
+        outputs=[single_question_image]
+    ).then(
+        fn=show_single_question_markdown,
+        inputs=[single_result_table],
+        outputs=[single_question_markdown]
+    )
+
 demo.launch()