Update app.py

app.py

@@ -34,9 +34,10 @@ def ensure_latex_packages():
 def generate_latex_response(image, question):
     """Génère une réponse en format LaTeX depuis le modèle Gemini."""
     try:
+        images = Image.open(image)
         response = client.models.generate_content(
             model="gemini-2.0-flash-thinking-exp",
-            contents=[image, question],
+            contents=[images, question],
         )
         # Extrait les pensées et la réponse
         thoughts = response.candidates[0].content.parts[0].text
@@ -94,8 +95,11 @@ def main():
     uploaded_file = st.file_uploader("Choisissez une image...", type=["png", "jpg", "jpeg"])
 
     if uploaded_file is not None:
+        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(uploaded_file)
+        image = Image.open(temp_image_path)
         st.image(image, caption="Image téléchargée", use_container_width=True)
 
         # Obtient la question de l'utilisateur
@@ -105,12 +109,9 @@ def main():
             if question:
                 with st.spinner("Génération de la réponse en cours..."):
                     # Convertit l'image PIL en bytes pour l'API Gemini
-                    bytes_data = io.BytesIO()
-                    image.save(bytes_data, format='PNG')
-                    image_bytes = bytes_data.getvalue()
-
+                    
                     # Génère la réponse LaTeX
-                    latex_response = generate_latex_response(image_bytes, question)
+                    latex_response = generate_latex_response(temp_image_path, question)
 
                     # Affiche la réponse LaTeX
                     st.markdown("### Code LaTeX généré:")