Update app.py

app.py

@@ -3,24 +3,38 @@ import pandas as pd
 from datetime import datetime, timedelta
 import io
 import os
-from huggingface_hub import HfApi, hf_hub_download
+from huggingface_hub import HfApi, hf_hub_download, login
 from typing import List, Tuple, Optional
 import tempfile
 import zipfile
 
 class DataManager:
-    def __init__(self):
+    def __init__(self, token=None):
         self.repo_id = "CIV3283/Data_Storage"
         self.branch = "data_branch"
-        self.api = HfApi()
+        
+        # 优先使用传入的token，否则从环境变量获取
+        self.token = token or os.getenv('CIV3283_access')
+        
+        # 如果有token，进行认证
+        if self.token:
+            try:
+                login(token=self.token)
+                print("✅ 认证成功")
+            except Exception as e:
+                print(f"❌ 认证失败: {e}")
+        
+        self.api = HfApi(token=self.token)
         
     def get_available_files(self) -> Tuple[List[str], List[str]]:
         """获取所有可用的feedback和query文件"""
         try:
-            # 获取仓库中的所有文件
+            # 尝试获取仓库中的所有文件
             repo_files = self.api.list_repo_files(
                 repo_id=self.repo_id,
-                revision=self.branch
+                revision=self.branch,
+                token=self.token,
+                repo_type="space"  # 明确指定这是一个space
             )
             
             feedback_files = []
@@ -35,8 +49,24 @@ class DataManager:
             return sorted(feedback_files), sorted(query_files)
             
         except Exception as e:
-            print(f"Error getting files: {e}")
-            return [], []
+            print(f"❌ 获取文件列表错误: {e}")
+            # 如果API调用失败，尝试手动生成可能的文件名列表
+            return self._generate_possible_filenames()
+    
+    def _generate_possible_filenames(self) -> Tuple[List[str], List[str]]:
+        """生成可能的文件名列表作为后备方案"""
+        feedback_files = []
+        query_files = []
+        
+        # 生成01到95的所有可能文件名
+        for i in range(1, 96):
+            student_id = f"{i:02d}"  # 格式化为两位数
+            feedback_file = f"CIV3283_Student_{student_id}_feedback_log.csv"
+            query_file = f"CIV3283_Student_{student_id}_query_log.csv"
+            feedback_files.append(feedback_file)
+            query_files.append(query_file)
+        
+        return feedback_files, query_files
     
     def download_and_read_csv(self, filename: str) -> pd.DataFrame:
         """下载并读取CSV文件"""
@@ -46,6 +76,8 @@ class DataManager:
                 repo_id=self.repo_id,
                 filename=filename,
                 revision=self.branch,
+                repo_type="space",
+                token=self.token,
                 cache_dir=tempfile.gettempdir()
             )
             
@@ -59,7 +91,8 @@ class DataManager:
             return df
             
         except Exception as e:
-            print(f"Error reading {filename}: {e}")
+            print(f"❌ 读取文件 {filename} 失败: {e}")
+            # 返回空DataFrame而不是完全失败
             return pd.DataFrame()
     
     def filter_data_by_date(self, df: pd.DataFrame, start_date: str, end_date: str) -> pd.DataFrame:
@@ -136,12 +169,26 @@ class DataManager:
         else:
             return "", "", "❌ 没有找到符合条件的数据"
 
-# 初始化数据管理器
-data_manager = DataManager()
+# 全局变量存储数据管理器
+data_manager = None
+
+def initialize_data_manager(hf_token=None):
+    """初始化数据管理器"""
+    global data_manager
+    data_manager = DataManager(token=hf_token)
+    return data_manager
 
-def process_and_download(file_type, start_date, end_date):
+def process_and_download(file_type, start_date, end_date, hf_token):
     """处理请求并返回下载文件"""
     try:
+        # 确保数据管理器已初始化
+        global data_manager
+        env_token = os.getenv('CIV3283_access')
+        current_token = hf_token or env_token
+        
+        if data_manager is None or data_manager.token != current_token:
+            data_manager = initialize_data_manager(current_token)
+        
         csv_content, filename, status = data_manager.process_files(file_type, start_date, end_date)
         
         if csv_content:
@@ -157,14 +204,30 @@ def process_and_download(file_type, start_date, end_date):
     except Exception as e:
         return f"❌ 处理过程中出现错误: {str(e)}", None, None
 
-def get_file_summary():
+def get_file_summary(hf_token=None):
     """获取文件概览"""
     try:
+        # 如果还没有初始化或token发生变化，重新初始化
+        global data_manager
+        env_token = os.getenv('CIV3283_access')
+        current_token = hf_token or env_token
+        
+        if data_manager is None or data_manager.token != current_token:
+            data_manager = initialize_data_manager(current_token)
+        
         feedback_files, query_files = data_manager.get_available_files()
         summary = f"📁 当前可用文件:\n"
         summary += f"   反馈文件: {len(feedback_files)} 个\n"
         summary += f"   查询文件: {len(query_files)} 个\n"
         summary += f"   总计: {len(feedback_files) + len(query_files)} 个文件"
+        
+        if env_token:
+            summary += f"\n✅ 使用环境变量 CIV3283_access 认证"
+        elif hf_token:
+            summary += f"\n✅ 使用手动输入的token认证"
+        else:
+            summary += f"\n⚠️  未找到认证token（可能无法访问私有仓库）"
+            
         return summary
     except Exception as e:
         return f"❌ 获取文件列表失败: {str(e)}"
@@ -175,18 +238,41 @@ def create_interface():
         gr.Markdown("# 🗃️ CIV3283 数据存储管理器")
         gr.Markdown("从 Hugging Face Space 的 data_branch 分支整理和下载学生数据")
         
+        # 认证部分
+        with gr.Row():
+            with gr.Column(scale=3):
+                hf_token = gr.Textbox(
+                    label="🔐 Hugging Face Token (可选)",
+                    placeholder="hf_xxxxxxxxxxxx",
+                    type="password",
+                    info="可选：手动输入token，留空将使用环境变量 CIV3283_access"
+                )
+            with gr.Column(scale=1):
+                gr.Markdown(f"""
+                **环境变量状态:**
+                {'✅ CIV3283_access 已设置' if os.getenv('CIV3283_access') else '❌ CIV3283_access 未找到'}
+                """)
+
+        
         # 文件概览
         with gr.Row():
             file_summary = gr.Textbox(
                 label="📊 文件概览",
-                value=get_file_summary(),
+                value="点击刷新按钮获取文件列表",
                 interactive=False,
-                max_lines=5
+                max_lines=6
             )
             refresh_btn = gr.Button("🔄 刷新文件列表", variant="secondary")
         
+        # 自动刷新一次（使用环境变量）
+        demo.load(
+            fn=lambda: get_file_summary(None),
+            outputs=file_summary
+        )
+        
         refresh_btn.click(
             fn=get_file_summary,
+            inputs=[hf_token],
             outputs=file_summary
         )
         
@@ -230,6 +316,11 @@ def create_interface():
         # 使用说明
         with gr.Accordion("📖 使用说明", open=False):
             gr.Markdown("""
+            ### 认证说明
+            - **环境变量优先**: 系统会自动使用环境变量 `CIV3283_access` 中的token
+            - **手动覆盖**: 如果需要使用不同的token，可以在上方手动输入
+            - **获取token**: 如果需要新token，访问 https://huggingface.co/settings/tokens
+            
             ### 功能说明
             1. **文件类型选择**: 选择要导出的数据类型
                - 查询文件: 包含搜索信息和查询响应
@@ -249,12 +340,13 @@ def create_interface():
             - 时间戳格式: 2025-06-16 08:55:33
             - 导出的数据保持原始顺序，不进行时间排序
             - 每次处理会显示详细的文件统计信息
+            - 如果无法访问仓库API，系统会尝试预设的文件名列表
             """)
         
         # 绑定处理函数
         process_btn.click(
             fn=process_and_download,
-            inputs=[file_type, start_date, end_date],
+            inputs=[file_type, start_date, end_date, hf_token],
             outputs=[status_box, download_file, gr.State()]
         )
     
@@ -264,7 +356,4 @@ def create_interface():
 if __name__ == "__main__":
     demo = create_interface()
     demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=True
     )