Update app.py

app.py

@@ -25,7 +25,7 @@ GOOGLE_API_KEY = os.environ.get("TOKEN")
 genai.configure(api_key=GOOGLE_API_KEY)
 
 # Fonction pour interroger Gemini
-def query_gemini(image_path, prompt="Résous ce problème mathématiques. Je veux qu'en réponse tu me donnes un rendu complet en utilisant du Latex."):
+def query_gemini(image_path,prompt_gemini):
     img = PIL.Image.open(image_path)
     model = genai.GenerativeModel(
         model_name="gemini-1.5-pro-002",
@@ -33,13 +33,13 @@ def query_gemini(image_path, prompt="Résous ce problème mathématiques. Je veu
         safety_settings=safety_settings
     )
     try:
-        response = model.generate_content([prompt, img], request_options={"timeout": 600})
+        response = model.generate_content([prompt_gemini, img], request_options={"timeout": 600})
         return response.text
     except Exception as e:
         return str(e)
 
 # Fonction pour interroger Qwen2
-def query_qwen2(image_path, question="Donne la réponse en français et en utilisant LaTeX."):
+def query_qwen2(image_path, question):
     try:
         client = Client("Qwen/Qwen2.5-Math-Demo")
         result = client.predict(
@@ -63,7 +63,10 @@ def upload_image():
 
     file = request.files['image']
     model_choice = request.form.get('model_choice', 'gemini')  # Obtient le choix du modèle
+    custom_instruction = request.form.get('custom_instruction')
 
+    prompt_gemini = f"Résous ce problème mathématiques. Je veux qu'en réponse tu me donnes un rendu complet en utilisant du Latex. {custom}"
+    question = f"Donne la réponse en français et en utilisant LaTeX.{custom_instruction}"
     if file.filename == '':
         return jsonify({'error': 'Aucun fichier sélectionné'}), 400