Update app.py

app.py

@@ -1,85 +1,70 @@
-from flask import Flask, request, render_template, jsonify
-import PIL.Image
-import google.generativeai as genai
+from flask import Flask, send_file, jsonify
+import fitz  # PyMuPDF pour manipuler le PDF
+import tempfile
 import os
-from tempfile import NamedTemporaryFile
-from gradio_client import Client, handle_file  # Importez gradio_client
 
 app = Flask(__name__)
 
-# Configuration de Gemini
-generation_config = {
-    "temperature": 1,
-    "max_output_tokens": 8192,
-}
+# Chemin vers le PDF source
+PDF_FILE = "static/pdf/large_file.pdf"
 
-safety_settings = [
-    {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_NONE"},
-    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_NONE"},
-    {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_NONE"},
-    {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_NONE"},
-]
+# Fonction pour générer une image temporaire d'une page du PDF
+def generate_temp_image(page_number, dpi):
+    try:
+        # Ouvre le PDF
+        with fitz.open(PDF_FILE) as doc:
+            # Vérifie que le numéro de page est valide
+            if page_number < 1 or page_number > len(doc):
+                raise ValueError(f"Numéro de page invalide : {page_number}")
 
-GOOGLE_API_KEY = os.environ.get("TOKEN")
+            # Extrait la page et génère une image
+            page = doc[page_number - 1]
+            pixmap = page.get_pixmap(dpi=dpi)
 
-genai.configure(api_key=GOOGLE_API_KEY)
+            # Crée un fichier temporaire pour stocker l'image
+            temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".jpg")
+            pixmap.save(temp_file.name)
+            return temp_file.name
+    except Exception as e:
+        raise RuntimeError(f"Erreur lors de la génération de l'image : {e}")
 
-# 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."):
-    img = PIL.Image.open(image_path)
-    model = genai.GenerativeModel(
-        model_name="gemini-1.5-pro-002",
-        generation_config=generation_config,
-        safety_settings=safety_settings
-    )
+# Endpoint pour récupérer une image basse résolution
+@app.route("/low_res/<int:page_number>")
+def get_low_res(page_number):
+    temp_path = None
     try:
-        response = model.generate_content([prompt, img], request_options={"timeout": 600})
-        return response.text
+        # Génère une image basse résolution
+        temp_path = generate_temp_image(page_number, dpi=72)
+        return send_file(temp_path, mimetype="image/jpeg")
     except Exception as e:
-        return str(e)
+        return jsonify({"error": str(e)}), 500
+    finally:
+        # Supprime le fichier temporaire après utilisation
+        if temp_path and os.path.exists(temp_path):
+            os.remove(temp_path)
 
-# Fonction pour interroger Qwen2
-def query_qwen2(image_path, question="Résous ce problème mathématiques. Donne la réponse en utilisant LaTeX."):
+# Endpoint pour récupérer une image haute résolution
+@app.route("/high_res/<int:page_number>")
+def get_high_res(page_number):
+    temp_path = None
     try:
-        client = Client("Qwen/Qwen2-Math-Demo")
-        result = client.predict(
-            image=handle_file(image_path),
-            sketchpad=None,
-            question=question,
-            api_name="/math_chat_bot"
-        )
-        return result
+        # Génère une image haute résolution
+        temp_path = generate_temp_image(page_number, dpi=300)
+        return send_file(temp_path, mimetype="image/jpeg")
     except Exception as e:
-        return str(e)
-
-@app.route('/')
-def index():
-    return render_template('math.html')
-
-@app.route('/upload', methods=['POST'])
-def upload_image():
-    if 'image' not in request.files:
-        return jsonify({'error': 'Aucune image fournie'}), 400
-
-    file = request.files['image']
-    model_choice = request.form.get('model_choice', 'gemini')  # Obtient le choix du modèle
+        return jsonify({"error": str(e)}), 500
+    finally:
+        # Supprime le fichier temporaire après utilisation
+        if temp_path and os.path.exists(temp_path):
+            os.remove(temp_path)
 
-    if file.filename == '':
-        return jsonify({'error': 'Aucun fichier sélectionné'}), 400
 
-    with NamedTemporaryFile(delete=False) as temp_file:
-        file.save(temp_file.name)
-        try:
-            if model_choice == "mariam's":
-                result = query_gemini(temp_file.name)
-            else:
-                result = query_qwen2(temp_file.name)
+from flask import render_template
 
-            os.unlink(temp_file.name)
-            return jsonify({'result': result, 'model': model_choice})
+@app.route("/viewer")
+def viewer():
+    return render_template("viewer.html")  # Assurez-vous que viewer.html est dans le dossier templates/
 
-        except Exception as e:
-            return jsonify({'error': str(e)}), 500
-            
-            
-  
+# Point d'entrée de l'application
+if __name__ == "__main__":
+    app.run(debug=True)