Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -5,8 +5,8 @@ import json
|
|
5 |
from flask import Flask, jsonify, render_template, request
|
6 |
from pydantic import BaseModel, Field
|
7 |
from typing import List, Optional
|
8 |
-
import psycopg2
|
9 |
-
from psycopg2.extras import RealDictCursor
|
10 |
from google import genai
|
11 |
from google.genai import types
|
12 |
from utils import load_prompt
|
@@ -32,13 +32,14 @@ class Dissertation(BaseModel):
|
|
32 |
parties: List[Partie]
|
33 |
conclusion: str = Field(description="La conclusion complète de la dissertation.")
|
34 |
|
35 |
-
# --- Configuration Gemini ---
|
36 |
try:
|
37 |
GOOGLE_API_KEY = os.environ.get("TOKEN")
|
38 |
if not GOOGLE_API_KEY:
|
39 |
logging.warning("La variable d'environnement TOKEN (GOOGLE_API_KEY) n'est pas définie.")
|
40 |
client = None
|
41 |
else:
|
|
|
42 |
client = genai.Client(api_key=GOOGLE_API_KEY)
|
43 |
except Exception as e:
|
44 |
logging.error(f"Erreur lors de l'initialisation du client GenAI: {e}")
|
@@ -54,7 +55,8 @@ SAFETY_SETTINGS = [
|
|
54 |
# --- Route Principale ---
|
55 |
@app.route('/')
|
56 |
def philosophie():
|
57 |
-
|
|
|
58 |
|
59 |
# --- Route API pour la génération de dissertation ---
|
60 |
@app.route('/api/generate_dissertation', methods=['POST'])
|
@@ -64,25 +66,40 @@ def generate_dissertation_api():
|
|
64 |
|
65 |
data = request.json
|
66 |
sujet = data.get('question', '').strip()
|
|
|
|
|
67 |
if not sujet:
|
68 |
return jsonify({"error": "Le champ 'question' est obligatoire."}), 400
|
|
|
|
|
|
|
69 |
|
70 |
try:
|
71 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
72 |
final_prompt = prompt_template.format(phi_prompt=sujet)
|
73 |
|
|
|
74 |
config = types.GenerateContentConfig(
|
75 |
safety_settings=SAFETY_SETTINGS,
|
76 |
response_mime_type="application/json",
|
77 |
response_schema=Dissertation,
|
78 |
)
|
79 |
|
|
|
80 |
response = client.models.generate_content(
|
81 |
-
model="gemini-2.5-flash",
|
82 |
contents=final_prompt,
|
83 |
config=config
|
84 |
)
|
85 |
|
|
|
86 |
if response.parsed:
|
87 |
return jsonify(response.parsed.dict())
|
88 |
else:
|
@@ -94,4 +111,4 @@ def generate_dissertation_api():
|
|
94 |
return jsonify({"error": f"Une erreur est survenue avec le service IA : {e}"}), 500
|
95 |
|
96 |
if __name__ == '__main__':
|
97 |
-
app.run(debug=True, port=
|
|
|
5 |
from flask import Flask, jsonify, render_template, request
|
6 |
from pydantic import BaseModel, Field
|
7 |
from typing import List, Optional
|
8 |
+
# import psycopg2 # La base de données n'est pas utilisée
|
9 |
+
# from psycopg2.extras import RealDictCursor
|
10 |
from google import genai
|
11 |
from google.genai import types
|
12 |
from utils import load_prompt
|
|
|
32 |
parties: List[Partie]
|
33 |
conclusion: str = Field(description="La conclusion complète de la dissertation.")
|
34 |
|
35 |
+
# --- Configuration Gemini (Ancien SDK) ---
|
36 |
try:
|
37 |
GOOGLE_API_KEY = os.environ.get("TOKEN")
|
38 |
if not GOOGLE_API_KEY:
|
39 |
logging.warning("La variable d'environnement TOKEN (GOOGLE_API_KEY) n'est pas définie.")
|
40 |
client = None
|
41 |
else:
|
42 |
+
# On utilise genai.Client comme dans votre code original
|
43 |
client = genai.Client(api_key=GOOGLE_API_KEY)
|
44 |
except Exception as e:
|
45 |
logging.error(f"Erreur lors de l'initialisation du client GenAI: {e}")
|
|
|
55 |
# --- Route Principale ---
|
56 |
@app.route('/')
|
57 |
def philosophie():
|
58 |
+
# Assurez-vous que le nom du template correspond à votre fichier HTML
|
59 |
+
return render_template("philosophie (13).html")
|
60 |
|
61 |
# --- Route API pour la génération de dissertation ---
|
62 |
@app.route('/api/generate_dissertation', methods=['POST'])
|
|
|
66 |
|
67 |
data = request.json
|
68 |
sujet = data.get('question', '').strip()
|
69 |
+
dissertation_type = data.get('type', 'type1').strip() # Récupère le type
|
70 |
+
|
71 |
if not sujet:
|
72 |
return jsonify({"error": "Le champ 'question' est obligatoire."}), 400
|
73 |
+
|
74 |
+
if dissertation_type not in ['type1', 'type2']:
|
75 |
+
return jsonify({"error": "Type de méthodologie invalide."}), 400
|
76 |
|
77 |
try:
|
78 |
+
# Sélection dynamique du fichier de prompt
|
79 |
+
prompt_filename = f"philo_dissertation_{dissertation_type}.txt"
|
80 |
+
prompt_template = load_prompt(prompt_filename)
|
81 |
+
|
82 |
+
if "Erreur:" in prompt_template:
|
83 |
+
logging.error(f"Fichier de prompt non trouvé : {prompt_filename}")
|
84 |
+
return jsonify({"error": "Configuration du prompt introuvable pour ce type."}), 500
|
85 |
+
|
86 |
final_prompt = prompt_template.format(phi_prompt=sujet)
|
87 |
|
88 |
+
# Configuration de la réponse structurée (comme dans votre code original)
|
89 |
config = types.GenerateContentConfig(
|
90 |
safety_settings=SAFETY_SETTINGS,
|
91 |
response_mime_type="application/json",
|
92 |
response_schema=Dissertation,
|
93 |
)
|
94 |
|
95 |
+
# Appel à l'API avec la syntaxe client.models.generate_content
|
96 |
response = client.models.generate_content(
|
97 |
+
model="gemini-2.5-flash", # Modèle performant compatible avec cette syntaxe
|
98 |
contents=final_prompt,
|
99 |
config=config
|
100 |
)
|
101 |
|
102 |
+
# Traitement de la réponse (comme dans votre code original)
|
103 |
if response.parsed:
|
104 |
return jsonify(response.parsed.dict())
|
105 |
else:
|
|
|
111 |
return jsonify({"error": f"Une erreur est survenue avec le service IA : {e}"}), 500
|
112 |
|
113 |
if __name__ == '__main__':
|
114 |
+
app.run(debug=True, port=5001)
|