| from flask import Flask, render_template_string, request, jsonify
|
| import os
|
|
|
|
|
| from parser import parse_all
|
| from data_layer import load_courses, filter_courses, recommend_courses, is_relevant
|
| from llm import answer, generate_recommendations
|
|
|
| app = Flask(__name__)
|
|
|
|
|
| courses = load_courses()
|
| print(f'Загружено курсов: {len(courses)}')
|
|
|
|
|
| HTML_TEMPLATE = '''
|
| <!DOCTYPE html>
|
| <html lang="ru">
|
| <head>
|
| <meta charset="UTF-8">
|
| <meta name="viewport" content="width=device-width, initial-scale=1.0">
|
| <title>ITMO Магистратура - Чат-бот</title>
|
| <style>
|
| body { font-family: Arial, sans-serif; max-width: 800px; margin: 0 auto; padding: 20px; }
|
| .chat-container { height: 300px; overflow-y: auto; border: 1px solid #ccc; padding: 10px; margin: 10px 0; }
|
| .message { margin: 5px 0; padding: 8px; border-radius: 5px; }
|
| .user { background: #e3f2fd; text-align: right; }
|
| .bot { background: #f5f5f5; }
|
| .input-group { display: flex; gap: 10px; margin: 10px 0; }
|
| input, button, select { padding: 8px; }
|
| button { background: #007bff; color: white; border: none; cursor: pointer; }
|
| .recommendations { margin-top: 20px; }
|
| textarea { width: 100%; height: 100px; }
|
| </style>
|
| </head>
|
| <body>
|
| <h1>🤖 ITMO Магистратура - Чат-бот</h1>
|
| <p>Загружено курсов: {{ courses_count }}</p>
|
|
|
| <div class="chat-container" id="chatContainer">
|
| <div class="message bot">Привет! Задавай вопросы о магистерских программах ITMO.</div>
|
| </div>
|
|
|
| <div class="input-group">
|
| <input type="text" id="messageInput" placeholder="Введите сообщение..." style="flex: 1;">
|
| <button onclick="sendMessage()">Отправить</button>
|
| <button onclick="clearChat()">Очистить</button>
|
| </div>
|
|
|
| <div class="recommendations">
|
| <h3>Рекомендации курсов</h3>
|
| <div class="input-group">
|
| <select id="semester">
|
| <option value="">Выберите семестр</option>
|
| <option value="1">1 семестр</option>
|
| <option value="2">2 семестр</option>
|
| <option value="3">3 семестр</option>
|
| <option value="4">4 семестр</option>
|
| </select>
|
| <button onclick="getRecommendations()">Получить рекомендации</button>
|
| <button onclick="updateData()">Обновить данные</button>
|
| </div>
|
| <textarea id="recommendationsOutput" placeholder="Здесь появятся рекомендации..." readonly></textarea>
|
| </div>
|
|
|
| <script>
|
| function addMessage(message, isUser = false) {
|
| const container = document.getElementById('chatContainer');
|
| const div = document.createElement('div');
|
| div.className = `message ${isUser ? 'user' : 'bot'}`;
|
| div.textContent = isUser ? 'Вы: ' + message : 'Бот: ' + message;
|
| container.appendChild(div);
|
| container.scrollTop = container.scrollHeight;
|
| }
|
|
|
| async function sendMessage() {
|
| const input = document.getElementById('messageInput');
|
| const message = input.value.trim();
|
| if (!message) return;
|
|
|
| addMessage(message, true);
|
| input.value = '';
|
|
|
| try {
|
| const response = await fetch('/api/chat', {
|
| method: 'POST',
|
| headers: {'Content-Type': 'application/json'},
|
| body: JSON.stringify({message: message})
|
| });
|
| const data = await response.json();
|
| addMessage(response.ok ? data.response : 'Ошибка: ' + data.error);
|
| } catch (error) {
|
| addMessage('Ошибка соединения: ' + error.message);
|
| }
|
| }
|
|
|
| async function getRecommendations() {
|
| const semester = document.getElementById('semester').value;
|
| if (!semester) {
|
| alert('Выберите семестр');
|
| return;
|
| }
|
|
|
| const output = document.getElementById('recommendationsOutput');
|
| output.value = 'Генерируем рекомендации...';
|
|
|
| try {
|
| const response = await fetch('/api/recommendations', {
|
| method: 'POST',
|
| headers: {'Content-Type': 'application/json'},
|
| body: JSON.stringify({
|
| programming_exp: 2,
|
| math_level: 2,
|
| interests: ['ml'],
|
| semester: semester
|
| })
|
| });
|
| const data = await response.json();
|
| output.value = response.ok ? data.response : 'Ошибка: ' + data.error;
|
| } catch (error) {
|
| output.value = 'Ошибка соединения: ' + error.message;
|
| }
|
| }
|
|
|
| async function updateData() {
|
| const output = document.getElementById('recommendationsOutput');
|
| output.value = 'Обновляем данные...';
|
|
|
| try {
|
| const response = await fetch('/api/update', {
|
| method: 'POST',
|
| headers: {'Content-Type': 'application/json'}
|
| });
|
| const data = await response.json();
|
| output.value = response.ok ? data.message : 'Ошибка: ' + data.error;
|
| if (response.ok) location.reload();
|
| } catch (error) {
|
| output.value = 'Ошибка соединения: ' + error.message;
|
| }
|
| }
|
|
|
| function clearChat() {
|
| document.getElementById('chatContainer').innerHTML =
|
| '<div class="message bot">Привет! Задавай вопросы о магистерских программах ITMO.</div>';
|
| }
|
|
|
| document.getElementById('messageInput').addEventListener('keypress', function(e) {
|
| if (e.key === 'Enter') sendMessage();
|
| });
|
| </script>
|
| </body>
|
| </html>
|
| '''
|
|
|
| @app.route('/')
|
| def index():
|
| """Главная страница"""
|
| return render_template_string(HTML_TEMPLATE, courses_count=len(courses))
|
|
|
| @app.route('/api/chat', methods=['POST'])
|
| def chat():
|
| """API для чата"""
|
| try:
|
| data = request.get_json()
|
| message = data.get('message', '').strip()
|
|
|
| if not message:
|
| return jsonify({'error': 'Пустое сообщение'}), 400
|
|
|
|
|
| if not is_relevant(message):
|
| response = '''Похоже, вопрос не относится к магистратурам ITMO и их учебным планам.
|
|
|
| Попробуйте спросить, например:
|
| • "Какие дисциплины по NLP в 1 семестре программы ИИ?"
|
| • "Расскажи о программе AI Product"
|
| • "Какие курсы по машинному обучению есть в программе ИИ?"
|
| • "Сколько кредитов за дисциплину 'Глубокое обучение'?"
|
| • "Какие курсы подходят для моего профиля?"'''
|
| return jsonify({'response': response})
|
|
|
|
|
| program_id = None
|
| message_lower = message.lower()
|
| if any(word in message_lower for word in ['ai product', 'продукт', 'менеджмент', 'аналитика']):
|
| program_id = 'ai_product'
|
| elif any(word in message_lower for word in ['ии', 'ai', 'машинное обучение', 'глубокое обучение', 'nlp', 'cv']):
|
| program_id = 'ai'
|
|
|
|
|
| semester = None
|
| for i in range(1, 5):
|
| if f'{i} семестр' in message_lower or f'{i} семестре' in message_lower:
|
| semester = i
|
| break
|
|
|
|
|
| context = filter_courses(message, program_id, semester)
|
|
|
| if not context:
|
| response = 'К сожалению, не нашел релевантной информации в учебных планах ITMO. Попробуйте переформулировать вопрос.'
|
| else:
|
|
|
| response = answer(message, context)
|
|
|
| return jsonify({'response': response})
|
|
|
| except Exception as e:
|
| return jsonify({'error': f'Ошибка обработки: {str(e)}'}), 500
|
|
|
| @app.route('/api/recommendations', methods=['POST'])
|
| def get_recommendations():
|
| """API для рекомендаций"""
|
| try:
|
| data = request.get_json()
|
| programming_exp = data.get('programming_exp', 2)
|
| math_level = data.get('math_level', 2)
|
| interests = data.get('interests', [])
|
| semester = data.get('semester', '')
|
|
|
| if not semester:
|
| return jsonify({'error': 'Пожалуйста, укажите семестр для получения рекомендаций.'}), 400
|
|
|
| try:
|
| semester_int = int(semester)
|
| except ValueError:
|
| return jsonify({'error': 'Пожалуйста, выберите корректный семестр.'}), 400
|
|
|
|
|
| profile = {
|
| 'programming_experience': programming_exp,
|
| 'math_level': math_level,
|
| 'interests': interests,
|
| 'semester': semester_int
|
| }
|
|
|
|
|
| recommended_courses = recommend_courses(profile)
|
|
|
| if not recommended_courses:
|
| return jsonify({'error': f'К сожалению, не найдено подходящих курсов для {semester} семестра.'}), 404
|
|
|
|
|
| response = generate_recommendations(recommended_courses, profile)
|
|
|
| return jsonify({'response': response})
|
|
|
| except Exception as e:
|
| return jsonify({'error': f'Ошибка получения рекомендаций: {str(e)}'}), 500
|
|
|
| @app.route('/api/update', methods=['POST'])
|
| def update_data():
|
| """API для обновления данных"""
|
| try:
|
| success = parse_all()
|
| if success:
|
|
|
| global courses
|
| courses = load_courses()
|
| return jsonify({'message': f'Данные успешно обновлены! Загружено {len(courses)} курсов.'})
|
| else:
|
| return jsonify({'error': 'Ошибка при обновлении данных. Используются базовые курсы.'}), 500
|
| except Exception as e:
|
| return jsonify({'error': f'Ошибка обновления данных: {str(e)}'}), 500
|
|
|
| @app.route('/api/status')
|
| def status():
|
| """API для статуса системы"""
|
| return jsonify({
|
| 'status': 'ok',
|
| 'courses_count': len(courses),
|
| 'llm_available': True
|
| })
|
|
|
|
|
| if __name__ == '__main__':
|
| port = int(os.environ.get('PORT', 7860))
|
| app.run(host='0.0.0.0', port=port, debug=False)
|
|
|
