Update app.py

app.py

@@ -36,6 +36,12 @@ def process_audio(file):
     audio = AudioSegment.from_file(file)
     return f"Аудио: длительность {len(audio) / 1000:.2f} секунд, частота {audio.frame_rate} Гц"
 
+# Функция для обработки текстового файла
+def process_txt(file):
+    with open(file.name, "r", encoding="utf-8") as f:
+        text = f.read()
+    return text
+
 # Функция для определения языка текста
 def detect_language(text):
     try:
@@ -64,6 +70,8 @@ def respond(
             file_info = process_video(file)
         elif file_type in ["mp3", "wav", "ogg"]:
             file_info = process_audio(file)
+        elif file_type == "txt":
+            file_info = process_txt(file)
         else:
             file_info = "Неизвестный тип файла"
         message += f"\n[Пользователь загрузил файл: {file.name}]\n{file_info}"
@@ -107,15 +115,30 @@ def respond(
 def reset_chat():
     return []
 
+# Функция для анализа текстового файла
+def analyze_txt(file):
+    if file is None:
+        return "Файл не загружен."
+    text = process_txt(file)
+    return f"Содержимое файла:\n{text}"
+
 # Создание интерфейса с использованием ChatInterface
 with gr.Blocks() as demo:
     gr.Markdown("# Felguk v0")
-    gr.Markdown("Чат-бот Felguk v0. Отвечает на том же языке, на котором вы написали. Задавайте вопросы и загружайте файлы (PDF, изображения, видео, аудио)!")
+    gr.Markdown("Чат-бот Felguk v0. Отвечает на том же языке, на котором вы написали. Задавайте вопросы и загружайте файлы (PDF, изображения, видео, аудио, txt)!")
     
     # Кнопка "Новый чат"
     with gr.Row():
         new_chat_button = gr.Button("Новый чат", variant="secondary")
 
+    # Felguk Tools: Txt Analyzer
+    with gr.Tab("Felguk Tools"):
+        gr.Markdown("## Txt Analyzer")
+        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)
+
     # Интерфейс чата
     chat_interface = gr.ChatInterface(
         respond,