Update app.py

app.py

@@ -1,13 +1,22 @@
-from flask import Flask, request, render_template
+from flask import Flask, request, render_template, send_from_directory
 import google.generativeai as genai
 import os
 from PIL import Image
 import io
+import subprocess
+import uuid
+import re
+import tempfile
 
 app = Flask(__name__)
+app.config['UPLOAD_FOLDER'] = 'uploads'
+os.makedirs(app.config['UPLOAD_FOLDER'], exist_ok=True)
 
 # Configuration de l'API Gemini
 token = os.environ.get("TOKEN")
+if not token:
+    raise ValueError("La variable d'environnement TOKEN doit être définie.")
+
 genai.configure(api_key=token)
 
 generation_config = {
@@ -34,6 +43,7 @@ model = genai.GenerativeModel(
 )
 
 
+
 @app.route("/", methods=["GET", "POST"])
 def index():
     e = ""
@@ -43,19 +53,62 @@ def index():
         else:
             image_file = request.files["image"]
             try:
-                image = Image.open(io.BytesIO(image_file.read()))
+                with tempfile.NamedTemporaryFile(delete=False) as temp_img:
+                    image_file.save(temp_img.name)
+                    image = Image.open(temp_img.name)
 
-                response = model.generate_content([mm, image])  # Passage de l'image
-                print(response.text)
-                e = response.text
+                response = model.generate_content([mm, image])
+                latex_code = response.text
+                os.remove(temp_img.name)
 
+                match = re.search(r"\\chemfig\{(.*?)\}", latex_code, re.DOTALL)
+                if match:
+                    chemfig_code = match.group(1)
+                    try:
+                        with tempfile.TemporaryDirectory() as tmpdirname:
+                            svg_filename = generate_svg_from_chemfig(chemfig_code, tmpdirname)
+                            e = f'<img src="/uploads/{svg_filename}" alt="Structure chimique">'
+                    except Exception as svg_error:
+                        e = f"Erreur lors de la génération de l'image SVG : {str(svg_error)}"
+                elif any(keyword in latex_code.lower() for keyword in ['.pdf', '.png', '.jpg', '.jpeg', '.gif', '.bmp', '.svg']):
+                    e = "Désolé, les images et les PDF ne sont pas encore pris en charge dans la réponse."
+                else:
+                    e = latex_code
 
-            except Exception as e: # gérer les erreurs potentielles d'ouverture de l'image
-                  e = f"Erreur lors du traitement de l'image : {str(e)}"
 
+            except Exception as img_err:
+                e = f"Erreur lors du traitement de l'image: {str(img_err)}"
 
     return render_template("index.html", e=e)
 
 
+def generate_svg_from_chemfig(chemfig_code, tmpdirname):
+    unique_id = str(uuid.uuid4())
+    tex_filename = os.path.join(tmpdirname, f"chem_{unique_id}.tex")
+    pdf_filename = os.path.join(tmpdirname, f"chem_{unique_id}.pdf")
+    svg_filename = f"chem_{unique_id}.svg"
+
+    tex_content = f"""\\documentclass[margin=10pt]{{standalone}}
+\\usepackage{{chemfig}}
+\\begin{{document}}
+\\chemfig{{{chemfig_code}}}
+\\end{{document}}"""
+
+    with open(tex_filename, "w") as tex_file:
+        tex_file.write(tex_content)
+
+    try:
+        subprocess.run(["pdflatex", "-interaction=nonstopmode", tex_filename], check=True, cwd=tmpdirname)
+        subprocess.run(["pdf2svg", pdf_filename, os.path.join(app.config['UPLOAD_FOLDER'], svg_filename)], check=True)
+        return svg_filename
+    except subprocess.CalledProcessError as e:
+        raise Exception(f"Erreur lors de la compilation LaTeX ou de la conversion en SVG : {e}")
+
+
+@app.route('/uploads/<filename>')
+def uploaded_file(filename):
+    return send_from_directory(app.config['UPLOAD_FOLDER'], filename)
+
+
 if __name__ == "__main__":
     app.run(debug=True)