Update src/streamlit_app.py

src/streamlit_app.py

@@ -119,51 +119,17 @@ def process_with_gemini(model, text, instruction="請解釋以下內容"):
         logger.error(f"Gemini處理失敗: {e}")
         return f"處理失敗: {str(e)}"
 
-def provide_download_options(result_df, export_format="嘗試標準CSV格式", include_checkbox=True):
-    """提供下載選項，根據選擇的匯出格式提供不同的CSV編碼方案"""
+def provide_download_options(result_df):
+    """提供下載選項，確保CSV中文不會出現亂碼"""
     
-    # 如果需要添加核取方塊欄位
-    if include_checkbox:
-        # 添加一個空的核取方塊欄位
-        result_df.insert(0, "☐", "")
-    
-    if export_format == "嘗試標準CSV格式":
-        # 標準方案：使用UTF-8-SIG編碼
-        csv_data = result_df.to_csv(index=False, encoding='utf-8-sig')
-        st.download_button(
-            label="下載結果為CSV",
-            data=csv_data.encode('utf-8-sig'),  # 確保編碼正確
-            file_name=f"pdf_process_result_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
-            mime="text/csv; charset=utf-8-sig",  # 指定MIME類型包含編碼
-        )
-    else:
-        # Excel兼容方案：使用GBK或Big5編碼，適合中文Windows環境
-        try:
-            # 先嘗試Big5編碼（適合繁體中文）
-            if "繁體中文" in target_language:
-                csv_data = result_df.to_csv(index=False, encoding='big5')
-                encoding_used = 'big5'
-            else:
-                # 否則嘗試GBK編碼（適合簡體中文）
-                csv_data = result_df.to_csv(index=False, encoding='gbk')
-                encoding_used = 'gbk'
-                
-            st.download_button(
-                label=f"下載結果為CSV (Excel兼容 - {encoding_used}編碼)",
-                data=csv_data,
-                file_name=f"pdf_process_result_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
-                mime=f"text/csv; charset={encoding_used}",
-            )
-        except UnicodeEncodeError:
-            # 如果特定編碼失敗，回退到UTF-8
-            st.warning("無法使用本地編碼，回退到UTF-8格式。若CSV開啟有亂碼，請用Excel的「資料匯入」功能匯入檔案。")
-            csv_data = result_df.to_csv(index=False, encoding='utf-8-sig')
-            st.download_button(
-                label="下載結果為CSV (UTF-8編碼)",
-                data=csv_data,
-                file_name=f"pdf_process_result_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
-                mime="text/csv; charset=utf-8-sig",
-            )
+    # 使用UTF-8-SIG編碼保證中文在CSV中不會出現亂碼
+    csv_data = result_df.to_csv(index=False, encoding='utf-8-sig')
+    st.download_button(
+        label="下載結果為CSV",
+        data=csv_data,
+        file_name=f"pdf_process_result_{datetime.now().strftime('%Y%m%d_%H%M%S')}.csv",
+        mime="text/csv; charset=utf-8-sig",  # 指定MIME類型包含編碼
+    )
     
     # 如果支援Excel格式，也提供Excel下載選項
     if EXCEL_SUPPORT:
@@ -182,17 +148,6 @@ def provide_download_options(result_df, export_format="嘗試標準CSV格式", i
 # 上傳PDF文件
 uploaded_file = st.file_uploader("上傳PDF文件", type="pdf")
 
-# 添加匯出選項
-st.sidebar.markdown("---")
-st.sidebar.markdown("### 匯出選項")
-export_format = st.sidebar.radio(
-    "CSV開啟後中文亂碼怎麼辦?", 
-    ["嘗試標準CSV格式", "使用Excel兼容格式"]
-)
-
-# 添加核取方塊選項
-include_checkbox = st.sidebar.checkbox("在CSV中添加核取方塊欄位", value=True, help="在Excel中開啟時，會顯示一個可勾選的核取方塊欄位")
-
 if uploaded_file is not None:
     # 顯示上傳成功訊息
     st.success(f"文件 '{uploaded_file.name}' 上傳成功!")
@@ -253,7 +208,7 @@ if uploaded_file is not None:
                             })
                             
                             # 提供下載選項
-                            provide_download_options(result_df, export_format)
+                            provide_download_options(result_df)
                         else:
                             # 先解釋後翻譯
                             with st.spinner("解釋內容..."):
@@ -282,6 +237,6 @@ if uploaded_file is not None:
                             })
                             
                             # 提供下載選項
-                            provide_download_options(result_df, export_format)
+                            provide_download_options(result_df)
             else:
                 st.error("無法提取PDF內容，請檢查文件格式是否正確。")