Update app.py

app.py

@@ -1,5 +1,4 @@
 import os
-import chardet
 import configparser
 from docx import Document
 import gradio as gr
@@ -8,39 +7,35 @@ from langchain_openai import AzureChatOpenAI
 # 讀取 config.ini 文件
 def load_config(config_file_path):
     config = configparser.ConfigParser()
-    
-    # 使用配置檔案路徑直接讀取文件
     try:
-        config.read(config_file_path)  # 使用檔案路徑來讀取
-        if not config.sections():  # 檢查是否包含區段標題
+        config.read(config_file_path)
+        if not config.sections():
             raise ValueError("config.ini 檔案缺少區段標題，請確認檔案格式正確。")
     except Exception as e:
         raise ValueError(f"讀取 config.ini 檔案時發生錯誤：{e}")
-    
     return config
 
 # 設定 Azure OpenAI 配置
 def create_openai_client(config):
-    return AzureChatOpenAI(
-        openai_api_version=config["AzureOpenAI"]["VERSION"],
-        azure_deployment=config["AzureOpenAI"]["DEPLOYMENT_NAME_GPT4o"],
-        azure_endpoint=config["AzureOpenAI"]["BASE"],
-        api_key=config["AzureOpenAI"]["KEY"],
-        temperature=0.2,
-    )
+    try:
+        return AzureChatOpenAI(
+            openai_api_version=config["AzureOpenAI"]["VERSION"],
+            azure_deployment=config["AzureOpenAI"]["DEPLOYMENT_NAME_GPT4o"],
+            azure_endpoint=config["AzureOpenAI"]["BASE"],
+            api_key=config["AzureOpenAI"]["KEY"],
+            temperature=0.2,
+        )
+    except KeyError as e:
+        raise ValueError(f"config.ini 檔案缺少必要的欄位：{e}")
 
 def read_docx(file):
     """Reads the content of a .docx file and returns the text."""
     doc = Document(file)
-    text = "\n".join([para.text for para in doc.paragraphs])
-    return text
+    return "\n".join([para.text for para in doc.paragraphs])
 
 def generate_monthly_report(weekly_reports_files, last_month_report_file, llm_gpt4o):
     """Generates a monthly report based on weekly reports and the last month's report."""
-    weekly_reports = []
-    for file in weekly_reports_files:
-        weekly_reports.append(read_docx(file))
-
+    weekly_reports = [read_docx(file) for file in weekly_reports_files]
     last_month_report = read_docx(last_month_report_file) if last_month_report_file else ""
 
     prompt = f"""請根據以下資訊生成本月月報：
@@ -54,23 +49,22 @@ def generate_monthly_report(weekly_reports_files, last_month_report_file, llm_gp
     prompt += """
     請將每個專案的進度內容進行詳細的描述，並參考上個月月報的格式,製作出本月的月報,並且描述下個月預計事項。
     """
-
-    # 使用 langchain 的 AzureChatOpenAI 生成月報
-    response = llm_gpt4o.invoke([("system", "You are a professional assistant."), ("human", prompt)])
-    return response.content
+    try:
+        response = llm_gpt4o.invoke([("system", "You are a professional assistant."), ("human", prompt)])
+        return response.content
+    except Exception as e:
+        raise ValueError(f"生成月報時發生錯誤：{str(e)}")
 
 def process_reports(weekly_reports_files, last_month_report_file, config_file):
     try:
-        # 將 config_file 參數傳遞為檔案路徑
         config_file_path = config_file.name
-        config = load_config(config_file_path)  # 讀取上傳的 config.ini
-        llm_gpt4o = create_openai_client(config)  # 創建 AzureOpenAI 客戶端
+        config = load_config(config_file_path)
+        llm_gpt4o = create_openai_client(config)
 
         if not weekly_reports_files or not last_month_report_file:
             return "請上傳所有必需的檔案（週報和上個月月報）。"
 
-        monthly_report = generate_monthly_report(weekly_reports_files, last_month_report_file, llm_gpt4o)
-        return monthly_report
+        return generate_monthly_report(weekly_reports_files, last_month_report_file, llm_gpt4o)
     except Exception as e:
         return f"生成月報時發生錯誤：{str(e)}"
 
@@ -82,11 +76,9 @@ with gr.Blocks() as demo:
     with gr.Row():
         weekly_reports_input = gr.File(label="上傳本月週報（多個 .docx 檔案）", file_types=[".docx"], file_count="multiple")
         last_month_report_input = gr.File(label="上傳上個月月報（單個 .docx 檔案）", file_types=[".docx"])
-
-    config_file_input = gr.File(label="上傳 config.ini 檔案")
+        config_file_input = gr.File(label="上傳 config.ini 檔案")
 
     output = gr.Textbox(label="生成的月報")
-
     submit_button = gr.Button("生成月報")
     submit_button.click(process_reports, inputs=[weekly_reports_input, last_month_report_input, config_file_input], outputs=output)