|
|
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)
|
|
|
|
