Create app.py

app.py

@@ -0,0 +1,79 @@
+import os
+import cv2
+import numpy as np
+from PIL import Image, ImageDraw
+import telebot
+import gradio as gr
+
+# Телеграм-токен
+API_TOKEN = '7458760921:AAFEMNVBuRcM_txnumnwl48u6MlKUS0J4YM'
+bot = telebot.TeleBot(API_TOKEN)
+
+# Создание папок для хранения файлов
+os.makedirs('uploads', exist_ok=True)
+os.makedirs('processed', exist_ok=True)
+
+# Загрузка классификатора для распознавания лиц
+face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')
+
+# Функция для обработки изображения
+def process_image(image_path):
+    img = cv2.imread(image_path)
+    gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
+    faces = face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30))
+    
+    pil_img = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
+    draw = ImageDraw.Draw(pil_img)
+    
+    coords = []
+    for (x, y, w, h) in faces:
+        draw.rectangle([x, y, x + w, y + h], outline="green", width=2)
+        coords.append((x, y, w, h))
+        draw.text((x, y - 10), f"({x}, {y})", fill="green")
+    
+    processed_path = image_path.replace('uploads', 'processed')
+    pil_img.save(processed_path)
+    return processed_path, coords
+
+# Обработчик сообщений с изображениями
+@bot.message_handler(content_types=['photo'])
+def handle_photo(message):
+    file_info = bot.get_file(message.photo[-1].file_id)
+    downloaded_file = bot.download_file(file_info.file_path)
+    
+    file_name = f"uploads/{message.photo[-1].file_id}.jpg"
+    with open(file_name, 'wb') as f:
+        f.write(downloaded_file)
+    
+    processed_path, coords = process_image(file_name)
+    
+    # Отправка обработанного изображения с координатами
+    with open(processed_path, 'rb') as f:
+        bot.send_photo(message.chat.id, f, caption=f"Координаты лиц: {coords}")
+
+# Gradio интерфейс
+def view_files():
+    files = os.listdir('processed')
+    return {file: f"processed/{file}" for file in files}
+
+def delete_file(file_name):
+    file_path = f"processed/{file_name}"
+    if os.path.exists(file_path):
+        os.remove(file_path)
+        return f"Файл {file_name} удалён."
+    return f"Файл {file_name} не найден."
+
+with gr.Blocks() as gr_interface:
+    gr.Markdown("## Управление обработанными изображениями")
+    with gr.Row():
+        gallery = gr.Gallery(label="Обработанные файлы").style(grid=3)
+    with gr.Row():
+        delete_button = gr.Button("Удалить выбранный файл")
+        delete_input = gr.Textbox(label="Имя файла для удаления")
+    delete_output = gr.Textbox(label="Результат удаления")
+
+    delete_button.click(delete_file, inputs=delete_input, outputs=delete_output)
+    gr.Interface(view_files, [], gallery).launch()
+
+# Запуск бота
+bot.polling()