Update src/streamlit_app.py

src/streamlit_app.py

@@ -10,38 +10,6 @@ from io import BytesIO
 import tempfile
 from pathlib import Path
 
-# 檢查是否有openpyxl套件，用於Excel匯出
-try:
-    import openpyxl
-    EXCEL_SUPPORT = True
-except ImportError:
-    EXCEL_SUPPORT = False
-
-
-# 如果支援Excel格式，也提供Excel下載選項
-if EXCEL_SUPPORT:
-    excel_buffer = BytesIO()
-    result_df.to_excel(excel_buffer, index=False, engine='openpyxl')
-    excel_data = excel_buffer.getvalue()
-    
-    st.download_button(
-        label="下載結果為Excel",
-        data=excel_data,
-        file_name=f"pdf_process_result_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xlsx",
-        mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
-        key="excel_download"  # 添加唯一key避免與CSV下載按鈕衝突
-    )
-# 需要安裝：pip install openpyxl
-excel_buffer = BytesIO()
-result_df.to_excel(excel_buffer, index=False, engine='openpyxl')
-excel_data = excel_buffer.getvalue()
-
-st.download_button(
-    label="下載結果為Excel",
-    data=excel_data,
-    file_name=f"pdf_process_result_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xlsx",
-    mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
-)
 # 設定頁面配置
 st.set_page_config(
     page_title="PDF 處理與翻譯工具",
@@ -59,6 +27,13 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+# 檢查是否有openpyxl套件，用於Excel匯出
+try:
+    import openpyxl
+    EXCEL_SUPPORT = True
+except ImportError:
+    EXCEL_SUPPORT = False
+
 # 應用標題
 st.title("PDF 內容處理與翻譯工具")
 st.markdown("使用 Gemini API 來處理和翻譯 PDF 文件內容")
@@ -145,6 +120,31 @@ def process_with_gemini(model, text, instruction="請解釋以下內容"):
         logger.error(f"Gemini處理失敗: {e}")
         return f"處理失敗: {str(e)}"
 
+def provide_download_options(result_df):
+    """提供下載選項"""
+    # 提供CSV下載選項
+    csv = result_df.to_csv(index=False)
+    st.download_button(
+        label="下載結果為CSV",
+        data=csv,
+        file_name=f"pdf_process_result_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
+        mime="text/csv",
+    )
+    
+    # 如果支援Excel格式，也提供Excel下載選項
+    if EXCEL_SUPPORT:
+        excel_buffer = BytesIO()
+        result_df.to_excel(excel_buffer, index=False, engine='openpyxl')
+        excel_data = excel_buffer.getvalue()
+        
+        st.download_button(
+            label="下載結果為Excel",
+            data=excel_data,
+            file_name=f"pdf_process_result_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xlsx",
+            mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+            key="excel_download"  # 添加唯一key避免與CSV下載按鈕衝突
+        )
+
 # 上傳PDF文件
 uploaded_file = st.file_uploader("上傳PDF文件", type="pdf")
 
@@ -206,6 +206,9 @@ if uploaded_file is not None:
                                 "原始內容": [selected_page_content],
                                 f"{target_language}翻譯": [translation]
                             })
+                            
+                            # 提供下載選項
+                            provide_download_options(result_df)
                         else:
                             # 先解釋後翻譯
                             with st.spinner("解釋內容..."):
@@ -232,14 +235,8 @@ if uploaded_file is not None:
                                 "內容解釋": [explanation],
                                 f"{target_language}翻譯": [translation]
                             })
-                        
-                        # 提供CSV下載選項
-                        csv = result_df.to_csv(index=False)
-                        st.download_button(
-                            label="下載結果為CSV",
-                            data=csv,
-                            file_name=f"pdf_process_result_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
-                            mime="text/csv",
-                        )
+                            
+                            # 提供下載選項
+                            provide_download_options(result_df)
             else:
                 st.error("無法提取PDF內容，請檢查文件格式是否正確。")