download_df_to_csv

assignment_ui.py

@@ -223,7 +223,7 @@ def create_assignment_ui(user_data, assignment_service, submission_service, dash
             def generate_submission_html(submission_json):
                 html_parts = []
                 
-                # 添���外层结构的开始标签
+                # 添加外层结构的开始标签
                 html_parts.append("""
                 <div style="background-color: #f8f9fa; padding: 30px; border-radius: 15px; font-family: 'Helvetica', sans-serif;">
                     <h2 style="color: #007bff; font-size: 28px; margin-bottom: 20px;">📝 學生回傳作業</h2>
@@ -545,13 +545,21 @@ def create_assignment_ui(user_data, assignment_service, submission_service, dash
                 assignments_df = gr.Dataframe(
                     label="作業列表",
                     wrap=True,
+                    interactive=False,
                 )
+            with gr.Row():
+                download_assignments_button = gr.Button("下載作業列表 CSV")
+                assignments_df_file = gr.File(height=10)
             
             with gr.Row():
                 submissions_df = gr.Dataframe(
                     label="所有提交的作業",
                     wrap=True,
+                    interactive=False,
                 )
+            with gr.Row():
+                download_submissions_button = gr.Button("下載提交作業 CSV")
+                submissions_df_file = gr.File(height=20)
 
             def update_dashboard(user_data):
                 # 獲取作業列表
@@ -564,10 +572,30 @@ def create_assignment_ui(user_data, assignment_service, submission_service, dash
 
                 return assignments_df, submissions_df
 
+            def download_df_to_csv(df):
+                import uuid
+                unique_filename = str(uuid.uuid4())
+                file_path = f"/tmp/{unique_filename}.csv"
+                df.to_csv(file_path, index=False)
+                return file_path
+
+
             refresh_dashboard_button.click(
                 fn=update_dashboard,
                 inputs=[user_data],
                 outputs=[assignments_df, submissions_df]
             )
 
+            download_assignments_button.click(
+                fn=download_df_to_csv,
+                inputs=[assignments_df],
+                outputs=[assignments_df_file]
+            )
+
+            download_submissions_button.click(
+                fn=download_df_to_csv,
+                inputs=[submissions_df],
+                outputs=[submissions_df_file]
+            )
+
     return assignment_interface