from flask import Flask, request, jsonify, render_template_string, session, redirect, url_for
import json
import os
from functools import wraps
from src.data_ingestion import ingest_data
from src.question_generation import QuestionGenerator
from src.grading import Grader
# ==================== CONFIGURAÇÃO ====================
CONFIG_FILE = os.path.join(os.path.dirname(__file__), "config.json")
def load_config():
try:
with open(CONFIG_FILE, "r") as f:
return json.load(f)
except (FileNotFoundError, json.JSONDecodeError):
return {}
config = load_config()
APP_PASSWORD = config.get("app_password")
SECRET_KEY = config.get("secret_key") or os.urandom(32)
# Inicializar app Flask
app = Flask(__name__)
app.secret_key = SECRET_KEY
# ==================== AUTENTICAÇÃO ====================
def login_required(f):
@wraps(f)
def wrapper(*args, **kwargs):
if not session.get("authenticated"):
return redirect(url_for("login", next=request.path))
return f(*args, **kwargs)
return wrapper
@app.route("/login", methods=["GET", "POST"])
def login():
global APP_PASSWORD
# Sempre recarrega o config.json, para suportar troca de senha sem reiniciar o app
config = load_config()
APP_PASSWORD = config.get("app_password")
error = None
if request.method == "POST":
senha_digitada = request.form.get("password", "")
if APP_PASSWORD and senha_digitada == APP_PASSWORD:
session["authenticated"] = True
return redirect(request.args.get("next") or url_for("index"))
error = "Senha incorreta. Tente novamente."
return render_template_string("""
Entrar
Acesso restrito
{% if error %}
{{error}}
{% endif %}
""", error=error)
@app.route("/logout")
def logout():
session.clear()
return redirect(url_for("login"))
# ==================== INICIALIZAR COMPONENTES ====================
try:
question_generator = QuestionGenerator()
except ValueError as e:
print(f"Warning: {e}")
question_generator = None
grader = Grader()
# ==================== PÁGINA INICIAL PROTEGIDA ====================
@app.route('/')
@login_required
def index():
return render_template_string("""
Sistema de Auto-Avaliação
Sistema de Auto-Avaliação
1. Gerar Perguntas
""")
# ==================== ROTAS PARA API ====================
@app.route('/generate_question', methods=['POST'])
def generate_question():
try:
data = request.json
text_content = data.get('text_content')
question_type = data.get('question_type', 'multiple_choice')
difficulty = data.get('difficulty', 'medium')
if not text_content:
return jsonify({'success': False, 'error': 'Text content is required'})
if question_generator is None:
return jsonify({'success': False, 'error': 'Question generator not available. Please set OPENAI_API_KEY environment variable.'})
question_json = question_generator.generate_question(text_content, question_type, difficulty)
if question_json:
question_data = json.loads(question_json)
return jsonify({'success': True, 'question': question_data})
else:
return jsonify({'success': False, 'error': 'Failed to generate question'})
except Exception as e:
return jsonify({'success': False, 'error': str(e)})
@app.route('/grade_question', methods=['POST'])
def grade_question():
try:
data = request.json
question = data.get('question')
student_answer = data.get('student_answer')
if not question or not student_answer:
return jsonify({'success': False, 'error': 'Question and student answer are required'})
grading_result = grader.grade_question(question, student_answer)
return jsonify({'success': True, 'grading': grading_result})
except Exception as e:
return jsonify({'success': False, 'error': str(e)})
# ==================== EXECUÇÃO ====================
if __name__ == '__main__':
app.run(debug=True, host='0.0.0.0', port=5000)