Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 import pandas as pd
 import io
-import re
 from typing import List, Optional, Tuple
 import logging
 
@@ -15,54 +14,41 @@ class WebsiteCategorizerApp:
         self.sheet_data = []
         self.current_index = 0
         self.categories = ["NEWS/BLOG", "E-commerce", "OTHER", "COMPANIES", "Short"]
-        self.results_data = []  # Для сохранения результатов
+        self.results_data = []
 
     def convert_google_sheet_url(self, sheet_url: str) -> str:
-        """Конвертирует URL Google таблицы в CSV экспорт URL"""
         try:
-            # Различные форматы URL Google таблиц
             if "/edit#gid=" in sheet_url:
-                csv_url = sheet_url.replace("/edit#gid=", "/export?format=csv&gid=")
+                return sheet_url.replace("/edit#gid=", "/export?format=csv&gid=")
             elif "/edit?usp=sharing" in sheet_url:
-                csv_url = sheet_url.replace("/edit?usp=sharing", "/export?format=csv")
+                return sheet_url.replace("/edit?usp=sharing", "/export?format=csv")
             elif "/edit" in sheet_url:
-                csv_url = sheet_url.replace("/edit", "/export?format=csv")
+                return sheet_url.replace("/edit", "/export?format=csv")
             else:
-                # Если URL уже в формате экспорта
-                csv_url = sheet_url
-
-            return csv_url
+                return sheet_url
         except Exception as e:
             logger.error(f"Ошибка конвертации URL: {e}")
             return ""
 
     def connect_to_sheet(self, sheet_url: str) -> Tuple[str, str]:
-        """Подключается к Google таблице через CSV экспорт"""
         try:
             if not sheet_url:
                 return "❌ Ошибка: Введите URL Google таблицы", ""
 
-            # Конвертируем URL в CSV формат
             csv_url = self.convert_google_sheet_url(sheet_url)
-
             if not csv_url:
                 return "❌ Ошибка: Неверный формат URL", ""
 
-            # Загружаем данные через pandas
             df = pd.read_csv(csv_url)
-
             if df.empty:
                 return "❌ Ошибка: Таблица пуста", ""
 
-            # Убедимся что есть минимум 2 столбца
             if len(df.columns) < 2:
                 return "❌ Ошибка: Нужно минимум 2 столбца (URL и категория)", ""
 
-            # Сохранение данных
             self.sheet_data = []
             self.results_data = []
 
-            # Получаем названия столбцов
             url_column = df.columns[0]
             category_column = df.columns[1]
 
@@ -76,8 +62,6 @@ class WebsiteCategorizerApp:
                         "url": url,
                         "category": category if category.lower() != 'nan' else ""
                     })
-
-                    # Добавляем в результаты
                     self.results_data.append({
                         "url": url,
                         "category": category if category.lower() != 'nan' else ""
@@ -89,288 +73,167 @@ class WebsiteCategorizerApp:
             self.current_index = 0
             self.sheet_url = sheet_url
 
-            success_msg = f"✅ Подключено успешно! Найдено {len(self.sheet_data)} записей"
-            first_url = self.get_current_url_for_display()
-
-            return success_msg, first_url
+            return f"✅ Подключено успешно! Найдено {len(self.sheet_data)} записей", self.get_current_url_for_display()
 
         except Exception as e:
             logger.error(f"Ошибка подключения к таблице: {e}")
-            error_msg = f"❌ Ошибка: {str(e)}"
-            error_msg += "\n\nУбедитесь что:\n- Таблица публичная (доступна всем по ссылке)\n- URL корректный"
-            return error_msg, ""
+            return f"❌ Ошибка: {str(e)}\n\nУбедитесь что таблица публичная и URL корректный", ""
 
     def get_current_url_for_display(self) -> str:
-        """Возвращает текущий URL для отображения в iframe"""
         if not self.sheet_data or self.current_index >= len(self.sheet_data):
             return ""
-
         url = self.sheet_data[self.current_index]["url"]
-
-        # Добавляем http:// если протокол не указан
         if url and not url.startswith(("http://", "https://")):
             url = "http://" + url
-
         return url
 
     def get_current_info(self) -> Tuple[str, str, str]:
-        """Возвращает информацию о текущей записи"""
         if not self.sheet_data:
             return "", "", "Нет данн��х"
-
         if self.current_index >= len(self.sheet_data):
             self.current_index = 0
-
         current = self.sheet_data[self.current_index]
-        url = current["url"]
-        category = current["category"]
-
-        info = f"{self.current_index + 1}/{len(self.sheet_data)}"
-
-        return url, category, info
+        return current["url"], current["category"], f"{self.current_index + 1}/{len(self.sheet_data)}"
 
     def navigate_to_index(self, index: int) -> Tuple[str, str, str, str]:
-        """Переходит к записи по индексу"""
         if not self.sheet_data:
             return "", "", "", "Нет данных"
-
-        # Ограничиваем индекс допустимыми значениями
         index = max(0, min(index, len(self.sheet_data) - 1))
         self.current_index = index
-
         url, category, info = self.get_current_info()
-        iframe_url = self.get_current_url_for_display()
-
-        return url, category, info, iframe_url
+        return url, category, info, self.get_current_url_for_display()
 
     def previous_record(self) -> Tuple[str, str, str, str]:
-        """Переход к предыдущей записи"""
         if not self.sheet_data:
             return "", "", "", "Нет данных"
-
-        if self.current_index > 0:
-            self.current_index -= 1
-        else:
-            self.current_index = len(self.sheet_data) - 1
-
+        self.current_index = (self.current_index - 1) % len(self.sheet_data)
         return self.navigate_to_index(self.current_index)
 
     def next_record(self) -> Tuple[str, str, str, str]:
-        """Переход к следующей записи"""
         if not self.sheet_data:
             return "", "", "", "Нет данных"
-
-        if self.current_index < len(self.sheet_data) - 1:
-            self.current_index += 1
-        else:
-            self.current_index = 0
-
+        self.current_index = (self.current_index + 1) % len(self.sheet_data)
         return self.navigate_to_index(self.current_index)
 
     def save_category(self, category: str) -> Tuple[str, str]:
-        """Сохраняет категорию локально и возвращает CSV для скачивания"""
         if not self.sheet_data:
             return "❌ Нет данных для сохранения", ""
-
         try:
-            # Обновляем локальные данные
             self.sheet_data[self.current_index]["category"] = category
             self.results_data[self.current_index]["category"] = category
-
-            # Создаем CSV файл с результатами
-            df_results = pd.DataFrame(self.results_data)
             csv_buffer = io.StringIO()
-            df_results.to_csv(csv_buffer, index=False, encoding='utf-8')
-            csv_content = csv_buffer.getvalue()
-
-            status_msg = f"✅ '{category}' сохранено"
-
-            return status_msg, csv_content
-
+            pd.DataFrame(self.results_data).to_csv(csv_buffer, index=False, encoding='utf-8')
+            return f"✅ '{category}' сохранено", csv_buffer.getvalue()
         except Exception as e:
             logger.error(f"Ошибка сохранения категории: {e}")
             return f"❌ Ошибка: {str(e)}", ""
 
     def export_results(self) -> str:
-        """Экспортирует все результаты в CSV"""
         if not self.results_data:
             return ""
-
-        df_results = pd.DataFrame(self.results_data)
         csv_buffer = io.StringIO()
-        df_results.to_csv(csv_buffer, index=False, encoding='utf-8')
+        pd.DataFrame(self.results_data).to_csv(csv_buffer, index=False, encoding='utf-8')
         return csv_buffer.getvalue()
 
-# Создаем экземпляр приложения
 app = WebsiteCategorizerApp()
 
-# Создание простого и надежного интерфейса
 with gr.Blocks(title="Категоризатор сайтов", theme=gr.themes.Soft()) as demo:
-
-    # Заголовок
-    gr.HTML("<h2 style='text-align: center; margin: 10px 0; color: #2d5aa0;'>🌐 Категоризатор сайтов</h2>")
-
-    with gr.Row():
-        # КОМПАКТНАЯ ЛЕВАЯ ПАНЕЛЬ (scale=1 - минимальный размер)
-        with gr.Column(scale=1):
-
-            # Подключение
-            gr.HTML("<h4 style='font-size: 14px; margin: 5px 0;'>📊 Подключение</h4>")
-            sheet_url_input = gr.Textbox(
-                label="URL Google таблицы",
-                placeholder="https://docs.google.com/spreadsheets/...",
-                lines=2
-            )
-            connect_btn = gr.Button("🔗 Подключить", variant="primary")
-            connection_status = gr.HTML("")
-
-            # Навигация
-            gr.HTML("<h4 style='font-size: 14px; margin: 5px 0;'>🧭 Навигация</h4>")
+    gr.HTML("<h2 style='text-align:center;'>🌐 Категоризатор сайтов</h2>")
+    with gr.Tabs():
+        with gr.TabItem("Категоризация"):
             with gr.Row():
-                prev_btn = gr.Button("⬅️", size="sm")
-                next_btn = gr.Button("➡️", size="sm")
-
-            record_info = gr.HTML("")
-            current_url_display = gr.Textbox(
-                label="Текущий URL", 
-                interactive=False,
-                lines=1
-            )
-
-            # Категория
-            gr.HTML("<h4 style='font-size: 14px; margin: 5px 0;'>🏷️ Категория</h4>")
-            category_dropdown = gr.Dropdown(
-                choices=app.categories,
-                label="Выберите категорию"
-            )
-            save_status = gr.HTML("")
-
-            # Экспорт
-            gr.HTML("<h4 style='font-size: 14px; margin: 5px 0;'>💾 Экспорт</h4>")
-            export_btn = gr.Button("📥 Скачать CSV")
-            export_file = gr.File(visible=False)
-
-        # БОЛЬШОЕ ОКНО ПРОСМОТРА (scale=5 - в 5 раз больше левой)
-        with gr.Column(scale=5):
-            gr.HTML("<h3 style='margin-bottom: 8px;'>🌍 Предварительный просмотр</h3>")
-
-            # Большое окно - 900px высота
-            website_viewer = gr.HTML(
-                value="""
-                <div style='height: 900px; display: flex; flex-direction: column; align-items: center; justify-content: center; 
-                           background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); 
-                           border: 2px dashed #ccc; border-radius: 12px; color: white; text-align: center;'>
-                    <h1 style='margin-bottom: 24px;'>🚀 Добро пожаловать!</h1>
-                    <p style='font-size: 20px; margin-bottom: 16px;'>Подключите Google таблицу</p>
-                    <p style='font-size: 16px; opacity: 0.8;'>Окно просмотра 900px высотой</p>
-                </div>
-                """
+                with gr.Column(scale=1):
+                    sheet_url_input = gr.Textbox(label="URL Google таблицы", lines=2)
+                    connect_btn = gr.Button("🔗 Подключить", variant="primary")
+                    connection_status = gr.HTML("")
+                    with gr.Row():
+                        prev_btn = gr.Button("⬅️", elem_id="prev-btn")
+                        next_btn = gr.Button("➡️", elem_id="next-btn")
+                    record_info = gr.HTML("")
+                    current_url_display = gr.Textbox(label="Текущий URL", interactive=False)
+                    category_dropdown = gr.Dropdown(choices=app.categories, label="Категория")
+                    save_status = gr.HTML("")
+                    export_btn = gr.Button("📥 Скачать CSV")
+                    export_file = gr.File(visible=False)
+                with gr.Column(scale=5):
+                    website_viewer = gr.HTML("""
+                        <div style='height:900px;display:flex;align-items:center;justify-content:center;background:#eee;border-radius:8px;'>
+                            <p>Подключите Google таблицу</p>
+                        </div>
+                    """)
+
+        with gr.TabItem("Текущая таблица"):
+            table_view = gr.DataFrame(
+                value=pd.DataFrame(app.results_data),
+                headers=["url", "category"],
+                datatype=["str", "str"],
+                interactive=True
             )
+            refresh_table_btn = gr.Button("🔄 Обновить таблицу")
 
-    # Скрытое состояние для CSV данных
     csv_data = gr.State("")
 
-    # Обработчики событий
     def handle_connect(url):
-        """Обработчик подключения к таблице"""
         status, iframe_url = app.connect_to_sheet(url)
-
-        if "✅" in status:  # Успешное подключение
+        if "✅" in status:
             url_display, category, info = app.get_current_info()
-
-            if iframe_url:
-                iframe_html = f'<iframe src="{iframe_url}" width="100%" height="900px" frameborder="0" style="border-radius: 8px;"></iframe>'
-            else:
-                iframe_html = "<div style='height: 900px; display: flex; align-items: center; justify-content: center; background: #f8d7da; border: 2px dashed #dc3545; border-radius: 8px; color: #721c24;'><p style='font-size: 18px;'>❌ URL не найден</p></div>"
-
-            info_html = f'<div style="color: green; background: #d4edda; padding: 8px; border-radius: 4px; margin: 4px 0;">{info}</div>'
-
-            return (
-                gr.HTML(f'<div style="color: green; background: #d4edda; padding: 8px; border-radius: 4px; margin: 4px 0;">✅ Готово!</div>'),
-                iframe_html,
-                url_display,
-                category,
-                info_html
-            )
-        else:  # Ошибка
-            return (
-                gr.HTML(f'<div style="color: #dc3545; background: #f8d7da; padding: 8px; border-radius: 4px; margin: 4px 0;">❌ Ошибка</div>'),
-                website_viewer.value,
-                "",
-                "",
-                gr.HTML("")
-            )
+            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_navigation(direction):
-        """Обработчик навигации"""
         if direction == "next":
             url_display, category, info, iframe_url = app.next_record()
         else:
             url_display, category, info, iframe_url = app.previous_record()
-
-        if iframe_url:
-            iframe_html = f'<iframe src="{iframe_url}" width="100%" height="900px" frameborder="0" style="border-radius: 8px;"></iframe>'
-        else:
-            iframe_html = "<div style='height: 900px; display: flex; align-items: center; justify-content: center; background: #f8d7da; border: 2px dashed #dc3545; border-radius: 8px; color: #721c24;'><p style='font-size: 18px;'>❌ URL не найден</p></div>"
-
-        info_html = f'<div style="color: green; background: #d4edda; padding: 8px; border-radius: 4px; margin: 4px 0;">{info}</div>'
-
-        return (
-            iframe_html,
-            url_display,
-            category,
-            info_html
-        )
+        iframe_html = f'<iframe src="{iframe_url}" width="100%" height="900px" style="border-radius:8px;"></iframe>'
+        return iframe_html, url_display, category, info
 
     def handle_category_change(category):
-        """Обработчик изменения категории"""
-        if category:
-            status, csv_content = app.save_category(category)
-            if "✅" in status:
-                status_html = f'<div style="color: green; background: #d4edda; padding: 8px; border-radius: 4px; margin: 4px 0;">{status}</div>'
-            else:
-                status_html = f'<div style="color: #dc3545; background: #f8d7da; padding: 8px; border-radius: 4px; margin: 4px 0;">{status}</div>'
-            return status_html, csv_content
-        return gr.HTML(""), ""
+        status, csv_content = app.save_category(category)
+        return status, csv_content
 
     def handle_export():
-        """Обработчик экспорта результатов"""
         csv_content = app.export_results()
         if csv_content:
-            # Сохраняем во временный файл
             with open("results.csv", "w", encoding="utf-8") as f:
                 f.write(csv_content)
             return gr.File(value="results.csv", visible=True)
         return gr.File(visible=False)
 
-    # Привязка событий
+    def refresh_table():
+        return pd.DataFrame(app.results_data)
+
     connect_btn.click(
         handle_connect,
         inputs=[sheet_url_input],
         outputs=[connection_status, website_viewer, current_url_display, category_dropdown, record_info]
     )
 
-    next_btn.click(
-        lambda: handle_navigation("next"),
-        outputs=[website_viewer, current_url_display, category_dropdown, record_info]
-    )
-
-    prev_btn.click(
-        lambda: handle_navigation("previous"),
-        outputs=[website_viewer, current_url_display, category_dropdown, record_info]
-    )
-
-    category_dropdown.change(
-        handle_category_change,
-        inputs=[category_dropdown],
-        outputs=[save_status, csv_data]
-    )
-
-    export_btn.click(
-        handle_export,
-        outputs=[export_file]
-    )
+    next_btn.click(lambda: handle_navigation("next"),
+                   outputs=[website_viewer, current_url_display, category_dropdown, record_info])
+    prev_btn.click(lambda: handle_navigation("previous"),
+                   outputs=[website_viewer, current_url_display, category_dropdown, record_info])
+    category_dropdown.change(handle_category_change,
+                              inputs=[category_dropdown],
+                              outputs=[save_status, csv_data])
+    export_btn.click(handle_export, outputs=[export_file])
+    refresh_table_btn.click(refresh_table, outputs=[table_view])
+
+    # JS для стрелок
+    gr.HTML("""
+    <script>
+    document.addEventListener('keydown', function(event) {
+        if (event.key === "ArrowRight") {
+            document.getElementById('next-btn')?.click();
+        }
+        if (event.key === "ArrowLeft") {
+            document.getElementById('prev-btn')?.click();
+        }
+    });
+    </script>
+    """)
 
-# Запуск приложения
 if __name__ == "__main__":
     demo.launch()