Upload 9 files

.gitattributes

@@ -1,35 +1,2 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,3 @@
+model_cache/
+models_cache/
+instance/

Dockerfile

@@ -0,0 +1,5 @@
+FROM python:3.9
+WORKDIR /app
+COPY . .
+RUN pip install -r requirements.txt
+CMD ["gunicorn", "--bind", "0.0.0.0:8080", "app:app"]

Procfile

@@ -0,0 +1 @@
+web: gunicorn app:app

app.py

@@ -0,0 +1,140 @@
+from flask import Flask, request, jsonify, render_template
+from transformers import pipeline
+import torch
+import torchaudio
+from pydub import AudioSegment
+import os
+import io
+os.environ["HF_HUB_DISABLE_SYMLINKS_WARNING"] = "1"
+from transformers import AutoModelForAudioClassification, AutoFeatureExtractor
+
+app = Flask(__name__)
+
+# Обновлённый словарь эмоций с surprise
+emotion_map = {
+    'joy': '😊 Радость',
+    'neutral': '😐 Нейтрально',
+    'anger': '😠 Злость',
+    'sadness': '😢 Грусть',
+    'surprise': '😲 Удивление'
+}
+
+# Добавим ffmpeg и ffprobe в PATH
+os.environ["PATH"] = r"C:\ffmpeg\bin;" + os.environ["PATH"]
+
+# Модель для текста
+text_classifier = pipeline("text-classification", model="cointegrated/rubert-tiny2-cedr-emotion-detection", top_k=None)
+
+# Модель для аудио
+# Аудио модель: универсальная, с поддержкой русского
+audio_classifier = pipeline(
+    "audio-classification",
+    model="superb/hubert-large-superb-er",
+    device=0 if torch.cuda.is_available() else -1
+)
+
+
+@app.route("/")
+def index():
+    return render_template("index.html")
+
+@app.route("/analyze", methods=["POST"])
+def analyze_text():
+    data = request.get_json()
+    text = data.get("text", "")
+    if not text.strip():
+        return jsonify({"error": "Пустой текст."}), 400
+
+    predictions = text_classifier(text)[0]
+    if not predictions:
+        return jsonify({"emotion": "❓ Не определено", "confidence": 0.0})
+
+    top_prediction = max(predictions, key=lambda x: x["score"])
+    label = top_prediction["label"]
+    confidence = round(top_prediction["score"], 2)
+
+    if confidence < 0.5:
+        return jsonify({"emotion": "🤔 Неуверенно", "confidence": confidence})
+
+    emotion = emotion_map.get(label, "❓ Неизвестно")
+    return jsonify({"emotion": emotion, "confidence": confidence})
+
+from pydub.utils import which
+
+# Проверка на наличие ffmpeg
+if not which("ffmpeg"):
+    print("FFmpeg не найден в системе!")
+
+
+@app.route('/analyze_audio', methods=['POST'])
+def analyze_audio():
+    try:
+        if 'audio' not in request.files:
+            return jsonify({'error': 'Аудио файл обязателен'}), 400
+
+        audio_file = request.files['audio']
+        temp_path = "temp_audio.wav"
+
+        # Конвертация в формат для модели
+        audio = AudioSegment.from_file(io.BytesIO(audio_file.read()))
+        audio = audio.set_frame_rate(16000).set_channels(1)
+        audio.export(temp_path, format="wav", codec="pcm_s16le")  # Явно указываем кодек
+
+        # Анализ
+        result = audio_classifier(temp_path)
+        os.remove(temp_path)
+
+        # Отладочный вывод
+        print("Raw model output:", result)
+
+        # Сопоставление с эмоциями
+        # Новый маппинг
+        emotion_mapping = {
+            'hap': 'happy',
+            'sad': 'sad',
+            'neu': 'neutral',
+            'ang': 'angry'
+        }
+
+        # Сбор эмоций
+        emotions = {v: 0.0 for v in emotion_mapping.values()}
+
+        for item in result:
+            label = item['label'].lower()
+            score = item['score']
+            if label in emotion_mapping:
+                key = emotion_mapping[label]
+                emotions[key] += score
+
+        # Нормализация
+        total = sum(emotions.values())
+        if total == 0:
+            return jsonify({'error': 'Не удалось распознать эмоции'}), 500
+
+        emotions = {k: round(v / total, 4) for k, v in emotions.items()}
+        dominant_emotion = max(emotions.items(), key=lambda x: x[1])
+
+        # Маппинг для ответа
+        response_map = {
+            'happy': '😊 Радость',
+            'sad': '😢 Грусть',
+            'angry': '😠 Злость',
+            'neutral': '😐 Нейтрально'
+        }
+
+        return jsonify({
+            'emotion': response_map.get(dominant_emotion[0], 'неизвестно'),
+            'confidence': dominant_emotion[1],
+            'details': emotions
+        })
+
+    except Exception as e:
+        print(f"Ошибка анализа аудио: {str(e)}")
+        return jsonify({'error': 'Ошибка обработки аудио'}), 500
+
+
+if __name__ == "__main__":
+    app.run(debug=True)
+
+if __name__ == "__main__":
+    app.run(debug=True)

requirements.txt

@@ -0,0 +1,7 @@
+flask>=2.0.0
+transformers>=4.15.0
+torch>=1.9.0
+torchaudio>=0.9.0
+pydub>=0.25.1
+librosa>=0.8.0
+numpy>=1.21.0

static/script.js

@@ -0,0 +1,133 @@
+document.addEventListener("DOMContentLoaded", () => {
+    // Объявляем переменные для записи аудио
+    let mediaRecorder;
+    let audioChunks = [];
+    let audioStream;
+
+    // Элементы интерфейса
+    const recordBtn = document.getElementById("record-btn");
+    const stopBtn = document.getElementById("stop-btn");
+    const sendBtn = document.getElementById("send-btn");
+    const uploadBtn = document.getElementById("upload-btn");
+    const userInput = document.getElementById("user-input");
+    const chatBox = document.getElementById("chat-box");
+    const audioFileInput = document.getElementById("audio-file");
+
+    // ===== 1. Отправка текста =====
+    sendBtn.addEventListener("click", async () => {
+        const text = userInput.value.trim();
+        if (!text) return;
+
+        appendMessage("user", text);
+        userInput.value = "";
+
+        try {
+            const response = await fetch("/analyze", {
+                method: "POST",
+                headers: { "Content-Type": "application/json" },
+                body: JSON.stringify({ text })
+            });
+
+            if (!response.ok) throw new Error("Ошибка сервера");
+
+            const data = await response.json();
+            appendMessage("bot", `Эмоция: ${data.emotion} (${(data.confidence * 100).toFixed(1)}%)`);
+        } catch (error) {
+            console.error("Ошибка:", error);
+            appendMessage("bot", `❌ Ошибка: ${error.message}`);
+        }
+    });
+
+    // ===== 2. Загрузка аудиофайла =====
+    uploadBtn.addEventListener("click", async () => {
+        const file = audioFileInput.files[0];
+        if (!file) return;
+
+        appendMessage("user", "Загружен аудиофайл...");
+
+        try {
+            const formData = new FormData();
+            formData.append("audio", file);
+
+            const response = await fetch("/analyze_audio", {
+                method: "POST",
+                body: formData
+            });
+
+            if (!response.ok) throw new Error("Ошибка сервера");
+
+            const data = await response.json();
+            appendMessage("bot", `Эмоция: ${data.emotion} (${(data.confidence * 100).toFixed(1)}%)`);
+        } catch (error) {
+            console.error("Ошибка:", error);
+            appendMessage("bot", `❌ Ошибка: ${error.message}`);
+        }
+    });
+
+    // ===== 3. Запись аудио =====
+    recordBtn.addEventListener("click", async () => {
+        try {
+            // Запрашиваем доступ к микрофону
+            audioStream = await navigator.mediaDevices.getUserMedia({ audio: true });
+
+            // Настраиваем запись
+            mediaRecorder = new MediaRecorder(audioStream);
+            audioChunks = [];
+
+            // Обработчики событий
+            mediaRecorder.ondataavailable = (e) => audioChunks.push(e.data);
+
+            mediaRecorder.onstop = async () => {
+                try {
+                    const audioBlob = new Blob(audioChunks, { type: "audio/wav" });
+                    appendMessage("user", "Отправлено голосовое сообщение...");
+
+                    const formData = new FormData();
+                    formData.append("audio", audioBlob, "recording.wav");
+
+                    const response = await fetch("/analyze_audio", {
+                        method: "POST",
+                        body: formData
+                    });
+
+                    if (!response.ok) throw new Error("Ошибка сервера");
+
+                    const data = await response.json();
+                    appendMessage("bot", `Эмоция: ${data.emotion} (${(data.confidence * 100).toFixed(1)}%)`);
+                } catch (error) {
+                    console.error("Ошибка:", error);
+                    appendMessage("bot", `❌ Ошибка: ${error.message}`);
+                } finally {
+                    // Освобождаем ресурсы
+                    audioStream.getTracks().forEach(track => track.stop());
+                }
+            };
+
+            mediaRecorder.start();
+            recordBtn.disabled = true;
+            stopBtn.disabled = false;
+
+        } catch (error) {
+            console.error("Ошибка записи:", error);
+            appendMessage("bot", "❌ Не удалось получить доступ к микрофону");
+        }
+    });
+
+    // Остановка записи
+    stopBtn.addEventListener("click", () => {
+        if (mediaRecorder && mediaRecorder.state === "recording") {
+            mediaRecorder.stop();
+            recordBtn.disabled = false;
+            stopBtn.disabled = true;
+        }
+    });
+
+    // ===== Вспомогательные функции =====
+    function appendMessage(sender, text) {
+        const message = document.createElement("div");
+        message.classList.add("message", sender === "user" ? "user-message" : "bot-message");
+        message.innerHTML = text; // Используем innerHTML для поддержки эмодзи
+        chatBox.appendChild(message);
+        chatBox.scrollTop = chatBox.scrollHeight;
+    }
+});

static/styles.css

@@ -0,0 +1,148 @@
+/* Основные стили (как у вас были) */
+body {
+    margin: 0;
+    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+    background: linear-gradient(135deg, #1e1e2f, #2a2a40);
+    height: 100vh;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    color: #fff;
+}
+
+.chat-container {
+    width: 500px;
+    background: #2a2a40;
+    border-radius: 16px;
+    box-shadow: 0 8px 16px rgba(0, 0, 0, 0.2);
+    overflow: hidden;
+    display: flex;
+    flex-direction: column;
+}
+
+.chat-title {
+    text-align: center;
+    font-size: 1.5rem;
+    color: #fff;
+    padding: 16px;
+    background: linear-gradient(90deg, #4a4ae8, #3b3b98);
+    margin: 0;
+    border-bottom: 1px solid rgba(255, 255, 255, 0.1);
+}
+
+.chat-box {
+    flex: 1;
+    padding: 16px;
+    overflow-y: auto;
+    background: #1e1e2f;
+    display: flex;
+    flex-direction: column;
+    gap: 12px;
+    max-height: 400px;
+}
+
+.message {
+    max-width: 80%;
+    padding: 12px 16px;
+    border-radius: 16px;
+    font-size: 0.9rem;
+    line-height: 1.4;
+    animation: fadeIn 0.3s ease-in-out;
+}
+
+.user-message {
+    background: linear-gradient(90deg, #4a4ae8, #3b3b98);
+    color: white;
+    align-self: flex-end;
+    border-bottom-right-radius: 0;
+}
+
+.bot-message {
+    background: #333;
+    color: #fff;
+    align-self: flex-start;
+    border-bottom-left-radius: 0;
+}
+
+@keyframes fadeIn {
+    from { opacity: 0; transform: translateY(10px); }
+    to { opacity: 1; transform: translateY(0); }
+}
+
+.input-container {
+    padding: 12px;
+    background: #2a2a40;
+    border-top: 1px solid rgba(255, 255, 255, 0.1);
+}
+
+.input-group {
+    display: flex;
+    gap: 8px;
+}
+
+.chat-input {
+    flex: 1;
+    padding: 12px;
+    border: 2px solid rgba(255, 255, 255, 0.2);
+    border-radius: 8px;
+    font-size: 0.9rem;
+    background: #1e1e2f;
+    color: #fff;
+}
+
+.chat-input:focus {
+    border-color: #4a4ae8;
+    outline: none;
+}
+
+.send-btn {
+    padding: 12px 20px;
+    background: linear-gradient(90deg, #4a4ae8, #3b3b98);
+    color: white;
+    border: none;
+    border-radius: 8px;
+    cursor: pointer;
+    font-size: 0.9rem;
+}
+
+/* Стили для аудио элементов */
+.audio-controls {
+    display: flex;
+    align-items: center;
+    gap: 10px;
+    padding: 12px;
+    background: #2a2a40;
+    border-top: 1px solid rgba(255, 255, 255, 0.1);
+}
+
+.audio-btn {
+    padding: 10px 15px;
+    background: linear-gradient(90deg, #4a4ae8, #3b3b98);
+    color: white;
+    border: none;
+    border-radius: 8px;
+    cursor: pointer;
+}
+
+.audio-btn:disabled {
+    background: #555;
+    cursor: not-allowed;
+}
+
+#waveform {
+    flex-grow: 1;
+    height: 60px;
+    background: #1e1e2f;
+    border-radius: 8px;
+}
+
+.file-upload {
+    display: flex;
+    gap: 10px;
+    align-items: center;
+    margin-top: 10px;
+}
+
+.file-upload input[type="file"] {
+    flex: 1;
+}

templates/index.html

@@ -0,0 +1,48 @@
+<!DOCTYPE html>
+<html lang="ru">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Emotion Analyzer</title>
+    <link rel="stylesheet" href="{{ url_for('static', filename='styles.css') }}">
+    <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.0.0/css/all.min.css">
+</head>
+<body>
+    <div class="chat-container">
+        <h1 class="chat-title">Анализатор эмоций</h1>
+
+        <div id="chat-box" class="chat-box">
+            <div class="message bot-message">
+                Привет! Отправьте текст или голосовое сообщение для анализа эмоций.
+            </div>
+        </div>
+
+        <div class="audio-controls">
+            <button id="record-btn" class="audio-btn">
+                <i class="fas fa-microphone"></i> Запись
+            </button>
+            <button id="stop-btn" class="audio-btn" disabled>
+                <i class="fas fa-stop"></i> Стоп
+            </button>
+            <canvas id="waveform" width="300" height="60"></canvas>
+        </div>
+
+        <div class="input-container">
+            <div class="input-group">
+                <input type="text" id="user-input" class="chat-input" placeholder="Введите сообщение...">
+                <button id="send-btn" class="send-btn">Отправить</button>
+            </div>
+        </div>
+
+        <div class="upload-container">
+            <label for="audio-file">Или загрузите аудиофайл:</label>
+            <input type="file" id="audio-file" accept="audio/*">
+            <button id="upload-btn" class="audio-btn">
+                <i class="fas fa-upload"></i> Загрузить
+            </button>
+        </div>
+    </div>
+
+    <script src="{{ url_for('static', filename='script.js') }}"></script>
+</body>
+</html>