Update app.py

app.py

@@ -122,43 +122,66 @@ def analyze_txt(file):
     text = process_txt(file)
     return f"Содержимое файла:\n{text}"
 
-# Создание интерфейса с использованием ChatInterface
+# Создание интерфейса
 with gr.Blocks() as demo:
     gr.Markdown("# Felguk v0")
-    gr.Markdown("Чат-бот Felguk v0. Отвечает на том же языке, на котором вы написали. Задавайте вопросы и загружайте файлы (PDF, изображения, видео, аудио, txt)!")
     
-    # Кнопка "Новый чат"
+    # Переменная для управления видимостью
+    current_tab = gr.State(value="chat")  # По умолчанию открыт чат
+
+    # Функция для переключения вкладок
+    def switch_tab(tab):
+        return tab
+
+    # Кнопки для переключения между вкладками
     with gr.Row():
+        chat_button = gr.Button("Чат", variant="primary")
+        tools_button = gr.Button("Felguk Tools", variant="secondary")
+
+    # Интерфейс чата
+    with gr.Column(visible=True) as chat_interface:
+        gr.Markdown("## Чат")
+        chat_interface = gr.ChatInterface(
+            respond,
+            additional_inputs=[
+                gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
+                gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
+                gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
+                gr.Slider(
+                    minimum=0.1,
+                    maximum=1.0,
+                    value=0.95,
+                    step=0.05,
+                    label="Top-p (nucleus sampling)",
+                ),
+                gr.File(label="Загрузите файл (опционально)"),  # Поле для загрузки файлов
+            ],
+        )
         new_chat_button = gr.Button("Новый чат", variant="secondary")
+        new_chat_button.click(fn=reset_chat, outputs=chat_interface.chatbot)
 
     # Felguk Tools: Txt Analyzer
-    with gr.Tab("Felguk Tools"):
-        gr.Markdown("## Txt Analyzer")
+    with gr.Column(visible=False) as tools_interface:
+        gr.Markdown("## Felguk Tools")
         txt_file = gr.File(label="Загрузите txt файл", file_types=[".txt"])
         txt_output = gr.Textbox(label="Содержимое файла", interactive=False)
         analyze_button = gr.Button("Анализировать")
         analyze_button.click(fn=analyze_txt, inputs=txt_file, outputs=txt_output)
+        back_to_chat_button = gr.Button("Вернуться в чат", variant="primary")
 
-    # Интерфейс чата
-    chat_interface = gr.ChatInterface(
-        respond,
-        additional_inputs=[
-            gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-            gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-            gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-            gr.Slider(
-                minimum=0.1,
-                maximum=1.0,
-                value=0.95,
-                step=0.05,
-                label="Top-p (nucleus sampling)",
-            ),
-            gr.File(label="Загрузите файл (опционально)"),  # Поле для загрузки файлов
-        ],
+    # Логика переключения вкладок
+    chat_button.click(
+        fn=lambda: (gr.Column.update(visible=True), gr.Column.update(visible=False)),
+        outputs=[chat_interface, tools_interface],
+    )
+    tools_button.click(
+        fn=lambda: (gr.Column.update(visible=False), gr.Column.update(visible=True)),
+        outputs=[chat_interface, tools_interface],
+    )
+    back_to_chat_button.click(
+        fn=lambda: (gr.Column.update(visible=True), gr.Column.update(visible=False)),
+        outputs=[chat_interface, tools_interface],
     )
-
-    # Привязка кнопки "Новый чат" к функции сброса истории
-    new_chat_button.click(fn=reset_chat, outputs=chat_interface.chatbot)
 
 # Запуск интерфейса
 if __name__ == "__main__":