Update app.py

app.py

@@ -1,3 +1,5 @@
+
+
 import streamlit as st
 from google import genai
 from PIL import Image
@@ -9,6 +11,7 @@ import tempfile
 import shutil
 
 # Authenticate using the application default credentials
+#credentials, project = google.auth.default()
 gen = api_key=os.environ['GOOGLE_API_KEY']
 # Initialize the Gemini client
 client = genai.Client(api_key=gen)
@@ -28,37 +31,34 @@ def ensure_latex_packages():
         st.error("pdflatex n'est pas trouvé. Veuillez installer TeX Live ou MiKTeX.")
         return False
 
-def generate_latex_response(image, question):
+def generate_latex_response(image_path, question):
     """Génère une réponse en format LaTeX depuis le modèle Gemini."""
     try:
-        images = Image.open(image)
+        with open(image_path, 'rb') as img_file:
+            image_data = img_file.read()
+            
         response = client.models.generate_content(
             model="gemini-2.0-flash-thinking-exp",
-            contents=[images, question],
+            contents=[image_data, question]
         )
-        # Extrait les pensées et la réponse
-        thoughts = response.candidates[0].content.parts[0].text
-        answer = response.candidates[0].content.parts[1].text
-
-        # Préparation du template LaTeX
-        latex_template = r"""\documentclass{article}
-\usepackage[utf8]{inputenc}
-\usepackage[T1]{fontenc}
-\usepackage{amsmath}
-\usepackage{amsfonts}
-\usepackage{amssymb}
-\begin{document}
-%s
-\end{document}
-""" % answer
         
-        return latex_template
-
+        # Extrait la réponse et retire les délimiteurs LaTeX
+        answer = response.text
+        
+        # Retire les délimiteurs ```latex au début et ``` à la fin
+        if answer.startswith("```latex"):
+            answer = answer[8:]  # Retire ```latex
+        if answer.endswith("```"):
+            answer = answer[:-3]  # Retire ```
+            
+        return answer.strip()  # Retire les espaces supplémentaires
+        
     except Exception as e:
-        return f"Une erreur s'est produite: {e}"
-
+        st.error(f"Erreur lors de la génération de la réponse: {str(e)}")
+        return None
 def latex_to_pdf(latex_content):
     """Convertit le contenu LaTeX en PDF."""
+    # Crée un dossier temporaire pour les fichiers LaTeX
     with tempfile.TemporaryDirectory() as temp_dir:
         # Chemin du fichier tex
         tex_file = os.path.join(temp_dir, "output.tex")
@@ -68,28 +68,25 @@ def latex_to_pdf(latex_content):
             f.write(latex_content)
         
         try:
-            # Execute pdflatex
-            process = subprocess.run(
-                ["pdflatex", "-interaction=nonstopmode", tex_file],
-                cwd=temp_dir,
-                capture_output=True,
-                text=True,
-                check=True
-            )
+            # Exécute pdflatex deux fois pour assurer la bonne génération des références
+            for _ in range(2):
+                subprocess.run(
+                    ["pdflatex", "-interaction=nonstopmode", "output.tex"],
+                    cwd=temp_dir,
+                    capture_output=True,
+                    check=True
+                )
+            
+            # Lit le fichier PDF généré
+            pdf_file = os.path.join(temp_dir, "output.pdf")
+            with open(pdf_file, "rb") as f:
+                pdf_data = f.read()
+            
+            return pdf_data
             
-            # Vérifie si le PDF existe
-            pdf_path = os.path.join(temp_dir, "output.pdf")
-            if os.path.exists(pdf_path):
-                with open(pdf_path, "rb") as f:
-                    return f.read()
-            else:
-                st.error("Le fichier PDF n'a pas été généré")
-                st.code(process.stdout)
-                return None
-                
         except subprocess.CalledProcessError as e:
-            st.error(f"Erreur lors de la compilation LaTeX: {str(e)}")
-            st.code(e.stdout)
+            st.error(f"Erreur lors de la génération du PDF: {e}")
+            st.code(e.output.decode())
             return None
 
 # Application Streamlit
@@ -106,7 +103,6 @@ def main():
         with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as tmp_file:
             tmp_file.write(uploaded_file.getvalue())
             temp_image_path = tmp_file.name
-
         # Affiche l'image téléchargée
         image = Image.open(temp_image_path)
         st.image(image, caption="Image téléchargée", use_container_width=True)
@@ -117,32 +113,29 @@ def main():
         if st.button("Générer la réponse"):
             if question:
                 with st.spinner("Génération de la réponse en cours..."):
+                    # Convertit l'image PIL en bytes pour l'API Gemini
+                    
                     # Génère la réponse LaTeX
                     latex_response = generate_latex_response(temp_image_path, question)
 
-                    # Affiche la réponse LaTeX brute
+                    # Affiche la réponse LaTeX
                     st.markdown("### Code LaTeX généré:")
                     st.text(latex_response)
 
-                    if latex_response:
-                        # Convertit en PDF
-                        with st.spinner("Conversion en PDF..."):
-                            pdf_data = latex_to_pdf(latex_response)
-                            
-                            if pdf_data:
-                                st.success("PDF généré avec succès!")
-                                # Crée un bouton de téléchargement
-                                st.download_button(
-                                    label="Télécharger le PDF",
-                                    data=pdf_data,
-                                    file_name="reponse_gemini.pdf",
-                                    mime="application/pdf"
-                                )
+                    # Convertit en PDF
+                    with st.spinner("Conversion en PDF..."):
+                        pdf_data = latex_to_pdf(latex_response)
+                        
+                        if pdf_data:
+                            # Crée un bouton de téléchargement
+                            st.download_button(
+                                label="Télécharger le PDF",
+                                data=pdf_data,
+                                file_name="reponse_gemini.pdf",
+                                mime="application/pdf"
+                            )
             else:
                 st.warning("Veuillez entrer une question.")
-        
-        # Nettoie le fichier temporaire
-        os.unlink(temp_image_path)
 
 if __name__ == "__main__":
     main()