Update app.py

app.py

@@ -51,6 +51,15 @@ def process_selected_videos(selected_videos):
     # 假資料模擬處理選擇的影片
     return f"已選擇的影片: {', '.join(selected_videos)}"
 
+def add_youtube_to_list(youtube_link, file_list):
+  if youtube_link:
+    file_list.append(youtube_link)
+    display_list = [os.path.basename(path) if os.path.basename(path) else path for path in file_list]
+  return gr.update(choices=display_list), ""
+
+def process_all_files(file_list):
+  return f"已處理的文件: {', '.join(file_list)}"
+
 with gr.Blocks() as demo:
     gr.Markdown("# AI Notes Assistant")
 
@@ -59,35 +68,38 @@ with gr.Blocks() as demo:
         chat_toggle = gr.Checkbox(label="顯示對話區域", value=True)
         feature_toggle = gr.Checkbox(label="顯示功能卡片", value=True)
 
-    with gr.Row(visible=True) as source_column:
-      with gr.Tab("檔案來源"):
-          file_list = gr.State([])
-          with gr.Row():
+    with gr.Column(visible=True) as source_column:
+        gr.Markdown("### 來源選單")
+        
+        file_list = gr.State([])
+        
+        with gr.Tab("上傳檔案"):
             upload_file = gr.File(label="從電腦添加文件", file_types=[".txt", ".pdf", ".docx"])
             add_file_button = gr.Button("添加到來源列表")
-          file_display = gr.CheckboxGroup(label="已上傳的文件", interactive=True)
-          add_file_button.click(add_to_file_list, inputs=[upload_file, file_list], outputs=[file_display, upload_file])
-          rag_button = gr.Button("處理選擇的文件")
-          rag_result = gr.Textbox(label="處理結果", interactive=False)
-          rag_button.click(process_selected_files, inputs=[file_display, file_list], outputs=[rag_result])
-
-      with gr.Tab("YouTube來源"):
-        gr.Markdown("### YouTube 播放清單")
-        youtube_data = get_youtube_playlist()
-        youtube_choices = format_youtube_choices(youtube_data)
-        youtube_videos = gr.CheckboxGroup(choices=youtube_choices, label="可選擇的影片", interactive=True)
-        process_videos_button = gr.Button("處理選擇的影片")
-        video_result = gr.Textbox(label="影片處理結果", interactive=False)
-        process_videos_button.click(process_selected_videos, inputs=[youtube_videos], outputs=[video_result])
+            add_file_button.click(add_to_file_list, inputs=[upload_file, file_list], outputs=[file_list, upload_file])
+    
+        with gr.Tab("YouTube 連結"):
+            youtube_link = gr.Textbox(label="輸入 YouTube 連結")
+            add_youtube_button = gr.Button("添加到來源列表")
+            add_youtube_button.click(add_youtube_to_list, inputs=[youtube_link, file_list], outputs=[file_list, youtube_link])
+
+        file_display = gr.CheckboxGroup(label="已上傳的文件", interactive=True)
+
+        process_files_button = gr.Button("處理檔案")
+        rag_result = gr.Textbox(label="處理結果", interactive=False)
+
+        process_files_button.click(process_all_files, inputs=[file_list], outputs=[rag_result])
+        file_list.change(lambda x: gr.update(choices = [os.path.basename(path) if os.path.basename(path) else path for path in x]), inputs=file_list, outputs=file_display)
+        
         
 
-    with gr.Row(visible=True) as chat_column:
+    with gr.Column(visible=True) as chat_column:
         gr.Markdown("### 對話區域")
         chatbot = gr.Chatbot(label="聊天記錄", type="messages")
         question = gr.Textbox(label="輸入問題，例如：文件的核心觀點是什麼？")
         ask_button = gr.Button("提問")
 
-    with gr.Row(visible=True) as feature_column:
+    with gr.Column(visible=True) as feature_column:
         gr.Markdown("### 功能卡片")
         with gr.Tab("摘要生成"):
             summary_button = gr.Button("生成摘要")