Update app.py

app.py

@@ -27,7 +27,7 @@ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 model = fasterrcnn_resnet50_fpn(weights="DEFAULT")
 model.eval()
 
-# Функция для обработки PDF
+# Функции для обработки данных (остаются без изменений)
 def process_pdf(file):
     pdf_reader = PyPDF2.PdfReader(file)
     text = ""
@@ -35,12 +35,10 @@ def process_pdf(file):
         text += page.extract_text()
     return text
 
-# Функция для обработки изображений
 def process_image(file):
     image = Image.open(file)
     return f"Изображение: {image.size[0]}x{image.size[1]} пикселей, формат: {image.format}"
 
-# Функция для обработки видео
 def process_video(file):
     cap = cv2.VideoCapture(file.name)
     frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
@@ -48,60 +46,45 @@ def process_video(file):
     cap.release()
     return f"Видео: длительность {duration:.2f} секунд, {frame_count} кадров"
 
-# Функция для обработки аудио
 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 remove_background(image):
     if image is None:
         return "**Ошибка:** Чем я буду удалять, если ты не загрузил изображение?"
     output = remove(image)
     return output
 
-# Функция для определения количества объектов на изображении
 def count_objects(image):
     if image is None:
         return "Изображение не загружено."
-    
-    # Преобразование изображения в формат, подходящий для модели
     img = Image.open(image.name).convert("RGB")
     img_tensor = F.to_tensor(img).unsqueeze(0)
-    
-    # Обнаружение объектов
     with torch.no_grad():
         predictions = model(img_tensor)
-    
-    # Подсчет количества объектов
     num_objects = len(predictions[0]['labels'])
     return f"Количество объектов на изображении: {num_objects}"
 
-# Функция для конвертации изображения
 def convert_image(image, target_format):
     if image is None:
         return None
     img = Image.open(image.name)
-    
-    # Создаем временный файл
     with tempfile.NamedTemporaryFile(delete=False, suffix=f".{target_format.lower()}") as tmp_file:
         img.save(tmp_file, format=target_format)
         return tmp_file.name
 
-# Функция для определения языка текста
 def detect_language(text):
     try:
         return detect(text)
     except:
-        return "en"  # По умолчанию английский
+        return "en"
 
-# Функция для обработки сообщений, истории и файлов
 def respond(
     message,
     history: list[tuple[str, str]],
@@ -111,7 +94,6 @@ def respond(
     top_p,
     file=None,
 ):
-    # Если загружен файл, обрабатываем его
     if file is not None:
         file_type = file.name.split(".")[-1].lower()
         if file_type == "pdf":
@@ -128,29 +110,21 @@ def respond(
             file_info = "Неизвестный тип файла"
         message += f"\n[Пользователь загрузил файл: {file.name}]\n{file_info}"
 
-    # Определяем язык сообщения
     language = detect_language(message)
-
-    # Добавляем системное сообщение с учетом языка
     if language == "ru":
         system_message = "Вы дружелюбный чат-бот, который понимает русский язык."
     else:
         system_message = "You are a friendly chatbot."
 
-    # Добавляем системное сообщение
     messages = [{"role": "system", "content": system_message}]
-
-    # Добавляем историю сообщений
     for val in history:
         if val[0]:
             messages.append({"role": "user", "content": val[0]})
         if val[1]:
             messages.append({"role": "assistant", "content": val[1]})
 
-    # Добавляем текущее сообщение пользователя
     messages.append({"role": "user", "content": message})
 
-    # Генерация ответа с использованием модели HuggingFaceH4/zephyr-7b-beta
     response = ""
     for message in client.chat_completion(
         messages,
@@ -163,31 +137,24 @@ def respond(
         response += token
         yield response
 
-# Функция для сброса истории чата
 def reset_chat():
     return []
 
-# Функция для анализа текстового файла
 def analyze_txt(file):
     if file is None:
         return "**Ошибка:** Ты не загрузил текстовый файл, а значит я не буду анализировать пустой файл."
     text = process_txt(file)
     return f"Содержимое файла:\n{text}"
 
-# Функция для изменения размера изображения
 def resize_image(image, width: int, height: int):
     if image is None:
         return None
-    
     img = Image.open(image.name)
     resized_img = img.resize((width, height))
-    
-    # Сохраняем изображение во временный файл
     with tempfile.NamedTemporaryFile(delete=False, suffix=".jpg") as tmp_file:
         resized_img.save(tmp_file.name)
         return tmp_file.name
 
-# Функция для перевода текста
 def translate_text(text: str, target_language: str):
     translator = Translator()
     try:
@@ -196,8 +163,63 @@ def translate_text(text: str, target_language: str):
     except Exception as e:
         return f"Ошибка перевода: {str(e)}"
 
-# Создание интерфейса
-with gr.Blocks() as demo:
+# Кастомные CSS-стили
+custom_css = """
+/* Основной фон */
+body, .gradio-container {
+    background-color: #f5f5f5;
+    font-family: 'Arial', sans-serif;
+}
+
+/* Заголовок */
+h1, h2, h3 {
+    color: #333;
+    font-weight: bold;
+}
+
+/* Кнопки */
+button {
+    background-color: #4CAF50;
+    color: white;
+    border: none;
+    padding: 10px 20px;
+    border-radius: 5px;
+    cursor: pointer;
+}
+
+button:hover {
+    background-color: #45a049;
+}
+
+/* Вкладки */
+.tab-button {
+    background-color: #e0e0e0;
+    color: #333;
+    border-radius: 5px 5px 0 0;
+}
+
+.tab-button.selected {
+    background-color: #4CAF50;
+    color: white;
+}
+
+/* Текстовые поля */
+input, textarea {
+    border: 1px solid #ccc;
+    border-radius: 5px;
+    padding: 10px;
+    width: 100%;
+}
+
+/* Изображения */
+img {
+    border-radius: 10px;
+    box-shadow: 0 4px 8px rgba(0, 0, 0, 0.1);
+}
+"""
+
+# Создание интерфейса с кастомными стилями
+with gr.Blocks(css=custom_css) as demo:
     # Заголовок с изображением
     gr.HTML("""
     <div style="text-align: center;">
@@ -208,20 +230,20 @@ with gr.Blocks() as demo:
     # Felguk News
     gr.Markdown("""
     ## Felguk News
-    **Последнее обновление:** 2023-10-25  
+    **Последнее обновление:** 2024-01-03
     - Добавлена поддержка новых форматов файлов.  
     - Улучшена производительность обработки изображений.  
     - Исправлены ошибки в переводе текста.  
     """)
 
     gr.Markdown("Чат-бот Felguk v0. Отвечает на том же языке, на котором вы написали. Задавайте вопросы и загружайте файлы (PDF, изображения, видео, аудио, txt)!")
-    
+
     # Кнопка "Новый чат"
     with gr.Row():
         new_chat_button = gr.Button("Новый чат", variant="secondary")
 
     # Felguk Tools: Отдельные инструменты
-    with gr.Tab("Felguk Tools"):
+    with gr.Tabs():
         # Вкладка Txt Analyzer
         with gr.Tab("Анализатор текста"):
             gr.Markdown("## Анализатор текста")
@@ -296,7 +318,7 @@ with gr.Blocks() as demo:
                 step=0.05,
                 label="Top-p (nucleus sampling)",
             ),
-            gr.File(label="Загрузите файл (опционально)"),  # Поле для загрузки файлов
+            gr.File(label="Загрузите файл (опционально)"),
         ],
     )