Update app.py

app.py

@@ -173,15 +173,29 @@ with gr.Blocks(title="Категоризатор сайтов", theme=gr.themes.
 
     csv_data = gr.State("")
 
+    # def handle_connect(url):
+    #     status, iframe_url = app.connect_to_sheet(url)
+    #     if "✅" in status:
+    #         url_display, category, info = app.get_current_info()
+    #         iframe_html = f'<iframe src="{iframe_url}" width="100%" height="900px" style="border-radius:8px;"></iframe>'
+    #         return status, iframe_html, url_display, category, info
+    #     else:
+    #         return status, website_viewer.value, "", "", ""
+
     def handle_connect(url):
         status, iframe_url = app.connect_to_sheet(url)
         if "✅" in status:
             url_display, category, info = app.get_current_info()
+    
+            # dynamically merge categories from data
+            all_categories = list(set(app.categories + [c for c in (row["category"] for row in app.results_data) if c]))
+            
             iframe_html = f'<iframe src="{iframe_url}" width="100%" height="900px" style="border-radius:8px;"></iframe>'
-            return status, iframe_html, url_display, category, info
+            return status, iframe_html, url_display, gr.Dropdown.update(choices=all_categories, value=category), info
         else:
-            return status, website_viewer.value, "", "", ""
+            return status, website_viewer.value, "", gr.Dropdown.update(choices=app.categories, value=None), ""
 
+    
     def handle_navigation(direction):
         if direction == "next":
             url_display, category, info, iframe_url = app.next_record()