added chat.py

app.py

@@ -1,5 +1,5 @@
 import os
-from flask import Flask, request, render_template, send_file, jsonify
+from flask import Flask, request, render_template, session, send_file, jsonify
 import io
 import requests
 import base64
@@ -8,14 +8,12 @@ from functions import (
     sentiment_analysis,
     generate_text,
     summarize_text,
-    # colorize_photo
+    chat_with_agent
 )
 
 
-
 app = Flask(__name__)
 
-
 UPLOAD_FOLDER = 'uploads'
 RESULTS_FOLDER = 'static/results'
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
@@ -52,6 +50,9 @@ def index():
             else:
                 error = "Введите текст для пересказа"
 
+        elif action=="chat":
+            pass
+
         # # ---------- Функции с фото ----------
         # elif action in ['colorize']:
         #     if 'image' not in request.files:
@@ -91,6 +92,19 @@ def index():
 
     return render_template('index.html', result_text=result_text, error=error)
 
+@app.route('/chat', methods=['POST'])
+def chat_endpoint():
+    """AJAX - эндпоинn для общения с ИИ - агентом."""
+    data = request.data.get_json()
+    user_message=data.get('message', '').strip()
+    if not user_message:
+        return jsonify({'error':'сообщение не молжет пустым'})
+
+    history = session.get('chat_history',[])
+    reply, new_history=chat_with_agent(history, user_message)
+    session['chat_history']=new_history
+    return jsonify({'reply':reply})
+
 if __name__=='__main__':
 
     app.run(debug=True, host='0.0.0.0', port=7860)

functions/__init__.py

@@ -1,4 +1,4 @@
 from functions.sentiment import sentiment_analysis
 from functions.generate_text import generate_text
 from functions.summarize import summarize_text
-
+from functions.chat import chat_with_agent

functions/generate_text.py

@@ -1,4 +1,4 @@
-from config import model_gt
+from networkbot.networkbot.config import model_gt
 import torch
 from transformers import GPT2Tokenizer, GPT2LMHeadModel
 

functions/sentiment.py

@@ -1,5 +1,5 @@
 
-from config import model_name
+from networkbot.networkbot.config import model_name
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 

functions/summarize.py

@@ -1,4 +1,4 @@
-from config import model_sum
+from networkbot.networkbot.config import model_sum
 from transformers import AutoTokenizer, T5ForConditionalGeneration
 import torch
 

requirements.txt

@@ -6,4 +6,6 @@ opencv-python
 gunicorn #синх
 pillow
 requests
+openai
+numpy
 

templates/index.html

@@ -27,46 +27,104 @@
                 <option value="sentiment">Анализ тональности</option>
                 <option value="generate">Генерация текста</option>
                 <option value="summarize">Пересказ текста</option>
-<!--                <option value="colorize">Раскрасить ч/б</option>-->
+                <option value="chat">Чат с ИИ-агентом</option>
             </select>
             <br><br>
             <textarea name="text" id="textInput" rows="4" cols="50" placeholder="Введите текст..."></textarea>
             <br>
-<!--            <input  type="file" name="image" accept="image/*" id="imageInput">-->
             <input type="submit" value="Отправить">
-
-<!--            <h3>Фото-функция</h3>-->
-<!--            <select name="action">-->
-<!--                <option value="colorize" selected>Раскрасить ч/б</option>-->
-<!--            </select>-->
-<!--            <br><br>-->
-<!--            <input  type="file" name="image" accept="image/*" required>-->
-<!--            <br>-->
-<!--            <input type="submit" value="Отправить">-->
         </div>
     </form>
 
-<!--    <script>-->
-<!--        const form = document.getElementById('mainForm');-->
-<!--        const actionSelect = document.getElementById('actionSelect');-->
-<!--        const textInput = document.getElementById('textInput');-->
-<!--        const imageInput = document.getElementById('imageInput');-->
+<script>
+    // Получаем историю из сессии (передана из Flask)
+    const initialHistory = {{ chat_history|tojson|safe }};
+
+    function addMessage(role, text) {
+        const container = document.getElementById('chat-messages');
+        const div = document.createElement('div');
+        div.className = 'message ' + role;
+        div.style.margin = '5px 0';
+        if (role === 'user') {
+            div.style.textAlign = 'right';
+            div.style.color = 'blue';
+        } else {
+            div.style.textAlign = 'left';
+            div.style.color = 'green';
+        }
+        div.textContent = text;
+        container.appendChild(div);
+        container.scrollTop = container.scrollHeight;
+    }
+
+    // Загружаем историю при загрузке страницы (если есть)
+    window.addEventListener('DOMContentLoaded', function() {
+        if (initialHistory) {
+            initialHistory.forEach(msg => {
+                if (msg.role !== 'system') {
+                    addMessage(msg.role, msg.content);
+                }
+            });
+        }
+        // Показываем чат, если выбрана опция
+        const select = document.getElementById('actionSelect');
+        toggleChat(select.value);
+    });
+
+    // Переключение видимости чата и текстового поля
+    function toggleChat(action) {
+        const textArea = document.getElementById('text-input-area');
+        const chatContainer = document.getElementById('chat-container');
+        const submitBtn = document.getElementById('submitBtn');
+        if (action === 'chat') {
+            textArea.style.display = 'none';
+            chatContainer.style.display = 'block';
+            submitBtn.style.display = 'none'; // скрываем кнопку отправки формы
+        } else {
+            textArea.style.display = 'block';
+            chatContainer.style.display = 'none';
+            submitBtn.style.display = 'inline-block';
+        }
+    }
+
+    document.getElementById('actionSelect').addEventListener('change', function() {
+        toggleChat(this.value);
+    });
+
+    // Отправка сообщения в чате
+    document.getElementById('chat-send').addEventListener('click', function() {
+        const input = document.getElementById('chat-input');
+        const msg = input.value.trim();
+        if (!msg) return;
+        addMessage('user', msg);
+        input.value = '';
+        // Отправляем на сервер
+        fetch('/chat', {
+            method: 'POST',
+            headers: {'Content-Type': 'application/json'},
+            body: JSON.stringify({message: msg})
+        })
+        .then(response => response.json())
+        .then(data => {
+            if (data.reply) {
+                addMessage('assistant', data.reply);
+            } else if (data.error) {
+                addMessage('assistant', 'Ошибка: ' + data.error);
+            }
+        })
+        .catch(error => {
+            addMessage('assistant', 'Ошибка сети: ' + error);
+        });
+    });
 
-<!--        form.addEventListener('submit', function(e) {-->
-<!--            const action = actionSelect.value;-->
-<!--            if (['sentiment', 'generate', 'summarize'].includes(action)) {-->
-<!--                if (!textInput.value.trim()) {-->
-<!--                    e.preventDefault();-->
-<!--                    alert('Введите текст');-->
-<!--                }-->
-<!--            } else if (['colorize',].includes(action)) {-->
-<!--                if (!imageInput.files.length) {-->
-<!--                    e.preventDefault();-->
-<!--                    alert('Выберите файл');-->
-<!--                }-->
-<!--            }-->
-<!--        });-->
-<!--    </script>-->
+    // Отправка по Enter
+    document.getElementById('chat-input').addEventListener('keypress', function(e) {
+        if (e.key === 'Enter') {
+            document.getElementById('chat-send').click();
+            e.preventDefault();
+        }
+    });
+</script>
 
     {% if result_text %}
         <div class="result">