Upload 18 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

.gitignore

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

.idea/.gitignore

@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml

.idea/emotion_analysis_system.iml

@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$">
+      <excludeFolder url="file://$MODULE_DIR$/.venv" />
+    </content>
+    <orderEntry type="jdk" jdkName="Python 3.11" jdkType="Python SDK" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="XmlDuplicatedId" enabled="false" level="ERROR" enabled_by_default="false" />
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="Black">
+    <option name="sdkName" value="Python 3.11 (emotion_analysis_system)" />
+  </component>
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.11" project-jdk-type="Python SDK" />
+  <component name="PyCharmProfessionalAdvertiser">
+    <option name="shown" value="true" />
+  </component>
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/emotion_analysis_system.iml" filepath="$PROJECT_DIR$/.idea/emotion_analysis_system.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>

.idea/workspace.xml

@@ -0,0 +1,75 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="AutoImportSettings">
+    <option name="autoReloadType" value="SELECTIVE" />
+  </component>
+  <component name="ChangeListManager">
+    <list default="true" id="80cad30e-20ea-4a3e-a71a-c380b9fb453b" name="Changes" comment="добавил gunicorn">
+      <change beforePath="$PROJECT_DIR$/.idea/vcs.xml" beforeDir="false" afterPath="$PROJECT_DIR$/.idea/vcs.xml" afterDir="false" />
+    </list>
+    <option name="SHOW_DIALOG" value="false" />
+    <option name="HIGHLIGHT_CONFLICTS" value="true" />
+    <option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" />
+    <option name="LAST_RESOLUTION" value="IGNORE" />
+  </component>
+  <component name="Git.Settings">
+    <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" />
+  </component>
+  <component name="ProjectColorInfo">{
+  &quot;associatedIndex&quot;: 7
+}</component>
+  <component name="ProjectId" id="2wCKFqS6j3fHR0o6mVerabcct5P" />
+  <component name="ProjectLevelVcsManager" settingsEditedManually="true" />
+  <component name="ProjectViewState">
+    <option name="hideEmptyMiddlePackages" value="true" />
+    <option name="showLibraryContents" value="true" />
+  </component>
+  <component name="PropertiesComponent"><![CDATA[{
+  "keyToString": {
+    "RunOnceActivity.OpenProjectViewOnStart": "true",
+    "RunOnceActivity.ShowReadmeOnStart": "true",
+    "git-widget-placeholder": "master",
+    "last_opened_file_path": "C:/Users/Айрат/Desktop/analyse_diplom_project-master",
+    "settings.editor.selected.configurable": "org.jetbrains.plugins.github.ui.GithubSettingsConfigurable"
+  }
+}]]></component>
+  <component name="SharedIndexes">
+    <attachedChunks>
+      <set>
+        <option value="bundled-python-sdk-09665e90c3a7-d3b881c8e49f-com.jetbrains.pycharm.community.sharedIndexes.bundled-PC-233.15026.15" />
+      </set>
+    </attachedChunks>
+  </component>
+  <component name="SpellCheckerSettings" RuntimeDictionaries="0" Folders="0" CustomDictionaries="0" DefaultDictionary="application-level" UseSingleDictionary="true" transferred="true" />
+  <component name="TaskManager">
+    <task active="true" id="Default" summary="Default task">
+      <changelist id="80cad30e-20ea-4a3e-a71a-c380b9fb453b" name="Changes" comment="" />
+      <created>1745540015058</created>
+      <option name="number" value="Default" />
+      <option name="presentableId" value="Default" />
+      <updated>1745540015058</updated>
+    </task>
+    <task id="LOCAL-00001" summary="добавил gunicorn">
+      <option name="closed" value="true" />
+      <created>1746432339386</created>
+      <option name="number" value="00001" />
+      <option name="presentableId" value="LOCAL-00001" />
+      <option name="project" value="LOCAL" />
+      <updated>1746432339386</updated>
+    </task>
+    <task id="LOCAL-00002" summary="добавил gunicorn">
+      <option name="closed" value="true" />
+      <created>1746433287813</created>
+      <option name="number" value="00002" />
+      <option name="presentableId" value="LOCAL-00002" />
+      <option name="project" value="LOCAL" />
+      <updated>1746433287813</updated>
+    </task>
+    <option name="localTasksCounter" value="3" />
+    <servers />
+  </component>
+  <component name="VcsManagerConfiguration">
+    <MESSAGE value="добавил gunicorn" />
+    <option name="LAST_COMMIT_MESSAGE" value="добавил gunicorn" />
+  </component>
+</project>

READme.txt

@@ -0,0 +1,16 @@
+Проблема с NumPy
+Переустановите NumPy и PyTorch:
+
+pip uninstall numpy torch -y
+pip install numpy torch --upgrade
+
+Конфликт версий PyTorch и NumPy
+Установите совместимые версии (например, для CPU):
+
+pip install numpy==1.23.5 torch==2.0.1 --upgrade
+
+Проблема с путями или кэшем
+Очистите кэш pip и переустановите зависимости:
+
+pip cache purge
+pip install -r requirements.txt --force-reinstall

app.py

@@ -0,0 +1,273 @@
+from flask import Flask, request, jsonify, render_template, session
+from transformers import pipeline
+import torch
+import torchaudio
+from pydub import AudioSegment
+import os
+import io
+import uuid
+from datetime import datetime
+import sqlite3
+from pathlib import Path
+import whisper  # Добавлена библиотека для преобразования речи в текст
+
+app = Flask(__name__)
+app.secret_key = 'your-very-secret-key-12345'
+
+
+# Инициализация БД
+def get_db_connection():
+    instance_path = Path('instance')
+    instance_path.mkdir(exist_ok=True)
+    db_path = instance_path / 'chats.db'
+    conn = sqlite3.connect(str(db_path))
+    conn.row_factory = sqlite3.Row
+    return conn
+
+
+def init_db():
+    conn = get_db_connection()
+    try:
+        conn.execute('''
+            CREATE TABLE IF NOT EXISTS chats (
+                chat_id TEXT PRIMARY KEY,
+                user_id TEXT,
+                created_at TEXT,
+                title TEXT
+            )
+        ''')
+        conn.execute('''
+            CREATE TABLE IF NOT EXISTS messages (
+                id INTEGER PRIMARY KEY AUTOINCREMENT,
+                chat_id TEXT,
+                sender TEXT,
+                content TEXT,
+                timestamp TEXT,
+                FOREIGN KEY(chat_id) REFERENCES chats(chat_id)
+            )
+        ''')
+        conn.commit()
+    finally:
+        conn.close()
+
+
+init_db()
+
+# Модели для анализа эмоций
+emotion_map = {
+    'joy': '😊 Радость',
+    'neutral': '😐 Нейтрально',
+    'anger': '😠 Злость',
+    'sadness': '😢 Грусть',
+    'surprise': '😲 Удивление'
+}
+
+# Инициализация модели для преобразования речи в текст
+try:
+    # Модель для преобразования речи в текст
+    speech_to_text_model = whisper.load_model("small")  # Можно использовать 'base' для меньшего потребления памяти
+
+    # Модели для анализа эмоций
+    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
+    )
+except Exception as e:
+    print(f"Ошибка загрузки моделей: {e}")
+    speech_to_text_model = None
+    text_classifier = None
+    audio_classifier = None
+
+
+def transcribe_audio(audio_path):
+    """Преобразование аудио в текст с помощью Whisper"""
+    if not speech_to_text_model:
+        return None
+
+    try:
+        result = speech_to_text_model.transcribe(audio_path, language="ru")
+        return result["text"]
+    except Exception as e:
+        print(f"Ошибка преобразования аудио в текст: {e}")
+        return None
+
+
+@app.route("/")
+def index():
+    if 'user_id' not in session:
+        session['user_id'] = str(uuid.uuid4())
+
+    conn = get_db_connection()
+    try:
+        chats = conn.execute(
+            "SELECT chat_id, title FROM chats WHERE user_id = ? ORDER BY created_at DESC",
+            (session['user_id'],)
+        ).fetchall()
+        return render_template("index.html", chats=chats)
+    finally:
+        conn.close()
+
+
+@app.route("/get_chats")
+def get_chats():
+    if 'user_id' not in session:
+        return jsonify([])
+
+    conn = get_db_connection()
+    try:
+        chats = conn.execute(
+            "SELECT chat_id, title FROM chats WHERE user_id = ? ORDER BY created_at DESC",
+            (session['user_id'],)
+        ).fetchall()
+        return jsonify([dict(chat) for chat in chats])
+    finally:
+        conn.close()
+
+
+@app.route("/start_chat", methods=["POST"])
+def start_chat():
+    if 'user_id' not in session:
+        session['user_id'] = str(uuid.uuid4())
+
+    chat_id = str(uuid.uuid4())
+    title = "Новый чат " + datetime.now().strftime("%d.%m %H:%M")
+
+    conn = get_db_connection()
+    try:
+        conn.execute(
+            "INSERT INTO chats (chat_id, user_id, created_at, title) VALUES (?, ?, datetime('now'), ?)",
+            (chat_id, session['user_id'], title)
+        )
+        conn.commit()
+        return jsonify({"chat_id": chat_id, "title": title})
+    finally:
+        conn.close()
+
+
+@app.route("/load_chat/<chat_id>", methods=["GET"])
+def load_chat(chat_id):
+    conn = get_db_connection()
+    try:
+        chat_exists = conn.execute(
+            "SELECT 1 FROM chats WHERE chat_id = ?", (chat_id,)
+        ).fetchone()
+
+        if not chat_exists:
+            return jsonify({"error": "Chat not found"}), 404
+
+        messages = conn.execute(
+            "SELECT sender, content FROM messages WHERE chat_id = ? ORDER BY timestamp ASC",
+            (chat_id,)
+        ).fetchall()
+
+        title_row = conn.execute(
+            "SELECT title FROM chats WHERE chat_id = ?", (chat_id,)
+        ).fetchone()
+
+        return jsonify({
+            "messages": [dict(msg) for msg in messages],
+            "title": title_row['title'] if title_row else "Без названия"
+        })
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+    finally:
+        conn.close()
+
+
+@app.route("/save_message", methods=["POST"])
+def save_message():
+    data = request.get_json()
+    if not all([data.get("chat_id"), data.get("sender"), data.get("content")]):
+        return jsonify({"error": "Missing parameters"}), 400
+
+    conn = get_db_connection()
+    try:
+        conn.execute(
+            "INSERT INTO messages (chat_id, sender, content, timestamp) VALUES (?, ?, ?, datetime('now'))",
+            (data['chat_id'], data['sender'], data['content'])
+        )
+        conn.commit()
+        return jsonify({"status": "success"})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+    finally:
+        conn.close()
+
+
+@app.route("/analyze", methods=["POST"])
+def analyze_text():
+    if not text_classifier:
+        return jsonify({"error": "Model not loaded"}), 500
+
+    text = request.get_json().get("text", "").strip()
+    if not text:
+        return jsonify({"error": "Empty text"}), 400
+
+    try:
+        predictions = text_classifier(text)[0]
+        top_prediction = max(predictions, key=lambda x: x["score"])
+        return jsonify({
+            "emotion": emotion_map.get(top_prediction["label"], "❓ Неизвестно"),
+            "confidence": round(top_prediction["score"], 2)
+        })
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+
+@app.route('/analyze_audio', methods=['POST'])
+def analyze_audio():
+    if not audio_classifier or not speech_to_text_model:
+        return jsonify({"error": "Model not loaded"}), 500
+
+    if 'audio' not in request.files:
+        return jsonify({'error': 'No audio file'}), 400
+
+    try:
+        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")
+
+        # Преобразование аудио в текст
+        transcribed_text = transcribe_audio(temp_path)
+
+        # Анализ эмоций в аудио
+        result = audio_classifier(temp_path)
+        os.remove(temp_path)
+
+        emotion_mapping = {
+            'hap': 'happy',
+            'sad': 'sad',
+            'neu': 'neutral',
+            'ang': 'angry'
+        }
+        emotions = {emotion_mapping.get(item['label'].lower(), 'neutral'): item['score']
+                    for item in result if item['label'].lower() in emotion_mapping}
+
+        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': round(dominant_emotion[1], 2),
+            'transcribed_text': transcribed_text if transcribed_text else "Не удалось распознать текст"
+        })
+    except Exception as e:
+        return jsonify({'error': str(e)}), 500
+
+
+if __name__ == "__main__":
+    app.run(debug=True)

requirements.txt

@@ -0,0 +1,9 @@
+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
+gunicorn
+openai-whisper

static/script.js

@@ -0,0 +1,233 @@
+document.addEventListener("DOMContentLoaded", () => {
+    let mediaRecorder, audioChunks = [], audioStream, currentChatId = null;
+    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");
+    const newChatBtn = document.getElementById("new-chat-btn");
+    const chatList = document.getElementById("chat-list");
+    const currentChatTitle = document.getElementById("current-chat-title");
+
+    // Инициализация при загрузке
+    initializeChats();
+
+    function initializeChats() {
+        const savedChatId = localStorage.getItem('currentChatId');
+
+        fetch("/get_chats")
+            .then(response => response.json())
+            .then(chats => {
+                renderChatList(chats);
+
+                if (savedChatId && chats.some(c => c.chat_id === savedChatId)) {
+                    loadChat(savedChatId);
+                } else if (chats.length > 0) {
+                    loadChat(chats[0].chat_id);
+                } else {
+                    startNewChat();
+                }
+            })
+            .catch(error => {
+                console.error("Ошибка загрузки чатов:", error);
+                startNewChat();
+            });
+    }
+
+    function renderChatList(chats) {
+        chatList.innerHTML = '';
+        chats.forEach(chat => {
+            const chatItem = document.createElement("div");
+            chatItem.className = "chat-item";
+            chatItem.dataset.chatId = chat.chat_id;
+            chatItem.innerHTML = `
+                <i class="fas fa-comment"></i>
+                <span class="chat-title">${chat.title}</span>
+            `;
+            chatItem.addEventListener("click", () => {
+                loadChat(chat.chat_id);
+                localStorage.setItem('currentChatId', chat.chat_id);
+            });
+            chatList.appendChild(chatItem);
+        });
+    }
+
+    newChatBtn.addEventListener("click", startNewChat);
+
+    function startNewChat() {
+        fetch("/start_chat", {
+            method: "POST",
+            headers: { "Content-Type": "application/json" },
+        })
+        .then(response => response.json())
+        .then(data => {
+            currentChatId = data.chat_id;
+            currentChatTitle.textContent = data.title;
+            chatBox.innerHTML = '<div class="message bot-message">Привет! Отправьте текст или голосовое сообщение для анализа эмоций.</div>';
+            initializeChats(); // Обновляем список чатов
+            localStorage.setItem('currentChatId', data.chat_id);
+        })
+        .catch(console.error);
+    }
+
+    function loadChat(chatId) {
+        fetch(`/load_chat/${chatId}`)
+        .then(response => response.json())
+        .then(data => {
+            if (data.error) throw new Error(data.error);
+
+            currentChatId = chatId;
+            currentChatTitle.textContent = data.title;
+            updateActiveChat(chatId);
+
+            chatBox.innerHTML = "";
+            data.messages.forEach(msg => {
+                appendMessage(msg.sender, msg.content);
+            });
+
+            localStorage.setItem('currentChatId', chatId);
+        })
+        .catch(error => {
+            console.error("Ошибка загрузки чата:", error);
+            appendMessage("bot", `❌ Ошибка: ${error.message}`);
+        });
+    }
+
+    function updateActiveChat(chatId) {
+        document.querySelectorAll(".chat-item").forEach(item => {
+            item.classList.toggle("active", item.dataset.chatId === chatId);
+        });
+    }
+
+    sendBtn.addEventListener("click", sendMessage);
+
+    async function sendMessage() {
+        const text = userInput.value.trim();
+        if (!text || !currentChatId) return;
+
+        appendAndSaveMessage("user", text);
+        userInput.value = "";
+
+        try {
+            const response = await fetch("/analyze", {
+                method: "POST",
+                headers: { "Content-Type": "application/json" },
+                body: JSON.stringify({ text })
+            });
+            const data = await response.json();
+            appendAndSaveMessage("bot", `Эмоция: ${data.emotion} (${(data.confidence * 100).toFixed(1)}%)`);
+        } catch (error) {
+            console.error("Ошибка:", error);
+            appendAndSaveMessage("bot", `❌ Ошибка: ${error.message}`);
+        }
+    }
+
+    uploadBtn.addEventListener("click", async () => {
+        const file = audioFileInput.files[0];
+        if (!file || !currentChatId) return;
+
+        appendAndSaveMessage("user", "Загружен аудиофайл...");
+
+        try {
+            const formData = new FormData();
+            formData.append("audio", file);
+
+            const response = await fetch("/analyze_audio", {
+                method: "POST",
+                body: formData
+            });
+            const data = await response.json();
+
+            // Добавляем распознанный текст в чат
+            if (data.transcribed_text) {
+                appendAndSaveMessage("user", `Распознанный текст: ${data.transcribed_text}`);
+            }
+
+            appendAndSaveMessage("bot", `Эмоция: ${data.emotion} (${(data.confidence * 100).toFixed(1)}%)`);
+        } catch (error) {
+            console.error("Ошибка:", error);
+            appendAndSaveMessage("bot", `❌ Ошибка: ${error.message}`);
+        }
+    });
+
+    recordBtn.addEventListener("click", startRecording);
+    stopBtn.addEventListener("click", stopRecording);
+
+    async function startRecording() {
+        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" });
+                    appendAndSaveMessage("user", "Отправлено голосовое сообщение...");
+
+                    const formData = new FormData();
+                    formData.append("audio", audioBlob, "recording.wav");
+
+                    const response = await fetch("/analyze_audio", {
+                        method: "POST",
+                        body: formData
+                    });
+                    const data = await response.json();
+
+                    // Добавляем распознанный текст в чат
+                    if (data.transcribed_text) {
+                        appendAndSaveMessage("user", `Распознанный текст: ${data.transcribed_text}`);
+                    }
+
+                    appendAndSaveMessage("bot", `Эмоция: ${data.emotion} (${(data.confidence * 100).toFixed(1)}%)`);
+                } catch (error) {
+                    console.error("Ошибка:", error);
+                    appendAndSaveMessage("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", "❌ Не удалось получить доступ к микрофону");
+        }
+    }
+
+    function stopRecording() {
+        if (mediaRecorder?.state === "recording") {
+            mediaRecorder.stop();
+            recordBtn.disabled = false;
+            stopBtn.disabled = true;
+        }
+    }
+
+    function appendMessage(sender, text) {
+        const message = document.createElement("div");
+        message.className = `message ${sender}-message`;
+        message.innerHTML = text;
+        chatBox.appendChild(message);
+        chatBox.scrollTop = chatBox.scrollHeight;
+    }
+
+    function appendAndSaveMessage(sender, text) {
+        appendMessage(sender, text);
+
+        if (currentChatId) {
+            fetch("/save_message", {
+                method: "POST",
+                headers: { "Content-Type": "application/json" },
+                body: JSON.stringify({
+                    chat_id: currentChatId,
+                    sender: sender,
+                    content: text
+                })
+            }).catch(console.error);
+        }
+    }
+});

static/styles.css

@@ -0,0 +1,228 @@
+/* Основные стили (как у вас были) */
+/* Основные стили */
+body {
+    margin: 0;
+    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
+    background: linear-gradient(135deg, #1e1e2f, #2a2a40);
+    height: 100vh;
+    color: #fff;
+    display: flex;
+}
+
+/* Сайдбар */
+.sidebar {
+    width: 250px;
+    background: #1a1a2e;
+    height: 100vh;
+    display: flex;
+    flex-direction: column;
+    border-right: 1px solid rgba(255, 255, 255, 0.1);
+}
+
+.sidebar-button {
+    padding: 12px 16px;
+    margin: 10px;
+    background: linear-gradient(90deg, #4a4ae8, #3b3b98);
+    color: white;
+    border: none;
+    border-radius: 8px;
+    cursor: pointer;
+    font-size: 0.9rem;
+    text-align: left;
+    display: flex;
+    align-items: center;
+    gap: 8px;
+}
+
+.sidebar-button:hover {
+    background: linear-gradient(90deg, #5a5aff, #4b4bb8);
+}
+
+.chat-list {
+    flex: 1;
+    overflow-y: auto;
+    padding: 0 10px;
+}
+
+.chat-item {
+    padding: 12px;
+    border-radius: 8px;
+    margin-bottom: 5px;
+    cursor: pointer;
+    display: flex;
+    align-items: center;
+    gap: 10px;
+    font-size: 0.9rem;
+}
+
+.chat-item:hover {
+    background: rgba(255, 255, 255, 0.1);
+}
+
+.chat-item.active {
+    background: rgba(74, 74, 232, 0.2);
+}
+
+.chat-title {
+    white-space: nowrap;
+    overflow: hidden;
+    text-overflow: ellipsis;
+}
+
+/* Основное содержимое */
+.main-content {
+    flex: 1;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    overflow-y: auto;
+}
+
+.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;
+    margin: 20px;
+}
+
+.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;
+}
+
+/* Красивая кнопка выбора файла */
+.custom-file-upload {
+    display: inline-block;
+    padding: 10px 16px;
+    background: linear-gradient(90deg, #4a4ae8, #3b3b98);
+    color: white;
+    font-size: 0.9rem;
+    border-radius: 8px;
+    cursor: pointer;
+    transition: background 0.3s ease;
+}
+
+.custom-file-upload:hover {
+    background: linear-gradient(90deg, #5a5aff, #4b4bb8);
+}
+
+/* Скрытие стандартного инпута */
+input[type="file"] {
+    display: none;
+}

templates/index.html

@@ -0,0 +1,68 @@
+<!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="sidebar">
+        <button id="new-chat-btn" class="sidebar-button">
+            <i class="fas fa-plus"></i> Новый чат
+        </button>
+        <div class="chat-list" id="chat-list">
+            {% for chat in chats %}
+            <div class="chat-item" data-chat-id="{{ chat.chat_id }}">
+                <i class="fas fa-comment"></i>
+                <span class="chat-title">{{ chat.title }}</span>
+            </div>
+            {% endfor %}
+        </div>
+    </div>
+
+    <div class="main-content">
+        <div class="chat-container">
+            <h1 class="chat-title" id="current-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>
+            </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" class="upload-label">Или загрузите аудиофайл:</label>
+                <div class="file-upload">
+                    <label for="audio-file" class="custom-file-upload">
+                        <i class="fas fa-file-audio"></i> Выбрать файл
+                    </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>
+        </div>
+    </div>
+
+    <script src="{{ url_for('static', filename='script.js') }}"></script>
+</body>
+</html>