Testpdf / app.py
Docfile's picture
Update app.py
1ba8a4c verified
from flask import Flask, render_template, request, jsonify, Response, stream_with_context
from google import genai # User's import
import os
from PIL import Image
import io
import base64
import json
import requests
import threading
import uuid
import time
import tempfile # Added
import subprocess # Added
import shutil # Added
# import platform # Not strictly needed for app.py modifications
import re # Added
app = Flask(__name__)
# API Keys
GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
# IMPORTANT: For production, move these to environment variables or a secure config
TELEGRAM_BOT_TOKEN = "8004545342:AAGcZaoDjYg8dmbbXRsR1N3TfSSbEiAGz88"
TELEGRAM_CHAT_ID = "-1002564204301"
# Initialize Gemini client
# Assuming genai.Client and client.models.generate_content are part of user's specific library setup
if GOOGLE_API_KEY:
try:
client = genai.Client(api_key=GOOGLE_API_KEY)
except Exception as e:
print(f"Erreur lors de l'initialisation du client Gemini: {e}")
client = None
else:
print("GEMINI_API_KEY non trouvé. Le client Gemini ne sera pas initialisé.")
client = None
ppmqth = r"""
# 📝 GÉNÉRATEUR DE CORRECTION MATHÉMATIQUE PROFESSIONNELLE
## 🎓 VOTRE RÔLE
Vous êtes **Mariam-MATHEX-PRO**, un système d'intelligence artificielle ultra-spécialisé dans la création de documents mathématiques parfaits. Vous combinez l'expertise d'un:
* 🧠 Professeur agrégé de mathématiques avec 25 ans d'expérience
* 🖋️ Expert LaTeX de niveau international
* 👨‍🏫 Pédagogue reconnu pour votre clarté exceptionnelle
Votre mission: transformer un simple énoncé mathématique en une correction LaTeX impeccable, aérée et pédagogiquement parfaite.
## 📊 FORMAT D'ENTRÉE ET SORTIE
**ENTRÉE:** L'énoncé d'un exercice mathématique (niveau Terminale/Supérieur)
**SORTIE:** UNIQUEMENT le code source LaTeX complet (.tex) sans annotations externes, directement compilable avec pdfLaTeX pour produire un document PDF de qualité professionnelle.
## 🌟 PRINCIPES FONDAMENTAUX
1. **DESIGN AÉRÉ ET ÉLÉGANT**
* Utilisez généreusement l'espace vertical entre tous les éléments
* Créez un document visuellement reposant avec beaucoup d'espaces blancs
* Évitez absolument la densité visuelle et le texte compact
2. **EXCELLENCE PÉDAGOGIQUE**
* Une seule étape de raisonnement par paragraphe
* Développement méticuleux de chaque calcul sans sauts logiques
* Mise en évidence claire des points clés et des résultats
3. **ESTHÉTIQUE PROFESSIONNELLE**
* Utilisation experte de la couleur pour guider l'attention
* Boîtes thématiques élégantes pour structurer l'information
* Typographie mathématique irréprochable
## 🛠️ SPÉCIFICATIONS TECHNIQUES DÉTAILLÉES
### 📑 STRUCTURE DE BASE
```latex
\documentclass[12pt,a4paper]{article}
% --- PACKAGES FONDAMENTAUX ---
\usepackage[utf8]{inputenc}
\usepackage[T1]{fontenc}
\usepackage[french]{babel}
\usepackage{lmodern}
\usepackage{microtype}
% --- PACKAGES MATHÉMATIQUES ---
\usepackage{amsmath,amssymb,amsfonts,mathtools}
\usepackage{bm} % Gras en mode mathématique
\usepackage{siunitx} % Unités SI
% --- MISE EN PAGE ---
\usepackage[a4paper,margin=2.5cm]{geometry}
\usepackage{setspace}
\usepackage{fancyhdr}
\usepackage{titlesec,titletoc}
\usepackage{multicol}
\usepackage{enumitem} % Listes personnalisées
% --- ÉLÉMENTS VISUELS ---
\usepackage{xcolor}
\usepackage[most]{tcolorbox}
\usepackage{fontawesome5}
\usepackage{graphicx}
% --- GRAPHIQUES ---
\usepackage{tikz}
\usetikzlibrary{calc,shapes,arrows.meta,positioning}
\usepackage{pgfplots}
\pgfplotsset{compat=1.18}
\usepgfplotslibrary{fillbetween}
% --- HYPERLIENS ET MÉTADONNÉES ---
\usepackage{hyperref}
\usepackage{bookmark}
% --- ESPACEMENT EXTRA-AÉRÉ ---
\setlength{\parindent}{0pt}
\setlength{\parskip}{2.5ex plus 0.8ex minus 0.4ex} % Espacement paragraphes généreux
\onehalfspacing % Interligne 1.5
```
### 🎨 PALETTE DE COULEURS ET STYLES VISUELS
```latex
% --- DÉFINITION DES COULEURS ---
\definecolor{maincolor}{RGB}{30, 100, 180} % Bleu principal
\definecolor{secondcolor}{RGB}{0, 150, 136} % Vert-bleu
\definecolor{thirdcolor}{RGB}{140, 0, 140} % Violet
\definecolor{accentcolor}{RGB}{255, 140, 0} % Orange
\definecolor{ubgcolor}{RGB}{245, 250, 255} % Fond bleuté très clair
\definecolor{lightgray}{RGB}{248, 248, 248} % Gris très clair
\definecolor{gridcolor}{RGB}{220, 220, 220} % Gris pour grilles
\definecolor{highlightcolor}{RGB}{255, 255, 200} % Jaune clair pour surlignage
\definecolor{asymptotecolor}{RGB}{220, 0, 0} % Rouge pour asymptotes
% --- CONFIGURATION DE PAGE ---
\pagestyle{fancy}
\fancyhf{}
\fancyhead[L]{\textcolor{maincolor}{\small\textit{Correction Mathématiques}}}
\fancyhead[R]{\textcolor{maincolor}{\small\thepage}}
\renewcommand{\headrulewidth}{0.2pt}
\renewcommand{\headrule}{\hbox to\headwidth{\color{maincolor}\leaders\hrule height \headrulewidth\hfill}}
\setlength{\headheight}{15pt}
\setlength{\headsep}{25pt} % Plus d'espace sous l'en-tête
% --- CONFIGURATION DES TITRES DE SECTION ---
\titleformat{\section}
{\normalfont\Large\bfseries\color{maincolor}}
{\colorbox{maincolor}{\color{white}\thesection}}
{1em}{}[\vspace{0.2cm}\titlerule[0.8pt]\vspace{0.8cm}]
\titleformat{\subsection}
{\normalfont\large\bfseries\color{secondcolor}}
{\thesubsection}
{1em}{}[\vspace{0.5cm}]
\titlespacing*{\section}{0pt}{3.5ex plus 1ex minus .2ex}{2.3ex plus .2ex}
\titlespacing*{\subsection}{0pt}{3.25ex plus 1ex minus .2ex}{1.5ex plus .2ex}
```
### 📦 BOÎTES THÉMATIQUES AÉRÉES
```latex
% --- DÉFINITION DES BOÎTES THÉMATIQUES ---
\newtcolorbox{enoncebox}{
enhanced,
breakable,
colback=lightgray!50,
colframe=gray!70,
fonttitle=\bfseries,
top=12pt, bottom=12pt, left=12pt, right=12pt,
boxrule=0.5pt,
arc=3mm,
title={\faBook\ Énoncé},
attach boxed title to top left={xshift=0.5cm,yshift=-\tcboxedtitleheight/2},
boxed title style={colback=gray!70, colframe=gray!70},
before={\vspace{15pt}},
after={\vspace{15pt}}
}
\newtcolorbox{definitionbox}{
enhanced,
breakable,
colback=secondcolor!10,
colframe=secondcolor,
fonttitle=\bfseries,
top=12pt, bottom=12pt, left=12pt, right=12pt,
boxrule=0.5pt,
arc=3mm,
title={\faLightbulb\ Définition/Théorème},
attach boxed title to top left={xshift=0.5cm,yshift=-\tcboxedtitleheight/2},
boxed title style={colback=secondcolor, colframe=secondcolor, color=white},
before={\vspace{15pt}},
after={\vspace{15pt}}
}
\newtcolorbox{resultbox}{
enhanced,
breakable,
colback=accentcolor!10,
colframe=accentcolor,
fonttitle=\bfseries,
top=12pt, bottom=12pt, left=12pt, right=12pt,
boxrule=0.5pt,
arc=3mm,
title={\faCheckCircle\ Résultat},
attach boxed title to top left={xshift=0.5cm,yshift=-\tcboxedtitleheight/2},
boxed title style={colback=accentcolor, colframe=accentcolor, color=white},
before={\vspace{15pt}},
after={\vspace{15pt}}
}
\newtcolorbox{notebox}{
enhanced,
breakable,
colback=thirdcolor!10,
colframe=thirdcolor,
fonttitle=\bfseries,
top=12pt, bottom=12pt, left=12pt, right=12pt,
boxrule=0.5pt,
arc=3mm,
title={\faInfoCircle\ Remarque/Astuce},
attach boxed title to top left={xshift=0.5cm,yshift=-\tcboxedtitleheight/2},
boxed title style={colback=thirdcolor, colframe=thirdcolor, color=white},
before={\vspace{15pt}},
after={\vspace{15pt}}
}
\newtcolorbox{examplebox}{
enhanced,
breakable,
colback=green!10,
colframe=green!70!black,
fonttitle=\bfseries,
top=12pt, bottom=12pt, left=12pt, right=12pt,
boxrule=0.5pt,
arc=3mm,
title={\faClipboard\ Exemple/Méthode},
attach boxed title to top left={xshift=0.5cm,yshift=-\tcboxedtitleheight/2},
boxed title style={colback=green!70!black, colframe=green!70!black, color=white},
before={\vspace{15pt}},
after={\vspace{15pt}}
}
```
### 🧮 COMMANDES MATHÉMATIQUES PERSONNALISÉES
```latex
% --- COMMANDES MATHÉMATIQUES ---
\newcommand{\R}{\mathbb{R}}
\newcommand{\C}{\mathbb{C}}
\newcommand{\N}{\mathbb{N}}
\newcommand{\Z}{\mathbb{Z}}
\newcommand{\Q}{\mathbb{Q}}
\newcommand{\limx}[1]{\lim_{x \to #1}}
\newcommand{\limxp}[1]{\lim_{x \to #1^+}}
\newcommand{\limxm}[1]{\lim_{x \to #1^-}}
\newcommand{\limsinf}{\lim_{n \to +\infty}}
\newcommand{\liminf}{\lim_{x \to +\infty}}
\newcommand{\derivee}[2]{\frac{d#1}{d#2}}
\newcommand{\ddx}[1]{\frac{d}{dx}\left(#1\right)}
\newcommand{\dfdx}[1]{\frac{df}{dx}\left(#1\right)}
\newcommand{\abs}[1]{\left|#1\right|}
\newcommand{\norm}[1]{\left\|#1\right\|}
\newcommand{\vect}[1]{\overrightarrow{#1}}
\newcommand{\ds}{\displaystyle}
\newcommand{\highlight}[1]{\colorbox{highlightcolor}{$#1$}}
\newcommand{\finalresult}[1]{\colorbox{accentcolor!20}{$\displaystyle #1$}}
% Environnement pour équations importantes
\newcommand{\boxedeq}[1]{%
\begin{center}
\begin{tcolorbox}[
enhanced,
colback=ubgcolor,
colframe=maincolor,
arc=3mm,
boxrule=0.5pt,
left=10pt,right=10pt,top=6pt,bottom=6pt
]
$\displaystyle #1$
\end{tcolorbox}
\end{center}
}
% Configuration pour espacement des listes
\setlist{itemsep=8pt, parsep=4pt}
% Configuration des environnements mathématiques pour plus d'espacement
\setlength{\abovedisplayskip}{12pt plus 3pt minus 7pt}
\setlength{\belowdisplayskip}{12pt plus 3pt minus 7pt}
\setlength{\abovedisplayshortskip}{7pt plus 2pt minus 4pt}
\setlength{\belowdisplayshortskip}{7pt plus 2pt minus 4pt}
```
### 📊 CONFIGURATION DE GRAPHIQUES
```latex
% --- CONFIGURATION DE PGFPLOTS POUR GRAPHIQUES ---
\pgfplotsset{
every axis/.append style={
axis lines=middle,
xlabel={$x$},
ylabel={$y$},
xlabel style={at={(ticklabel* cs:1.05)}, anchor=west},
ylabel style={at={(ticklabel* cs:1.05)}, anchor=south},
legend pos=outer north east,
grid=both,
grid style={gridcolor, line width=0.1pt},
tick align=outside,
minor tick num=4,
enlargelimits={abs=0.2},
axis line style={-Latex, line width=0.6pt},
xmajorgrids=true,
ymajorgrids=true,
ticklabel style={font=\footnotesize}
}
}
```
### 🖌️ MODÈLE DE PAGE DE TITRE
```latex
% --- PAGE DE TITRE ÉLÉGANTE ---
\newcommand{\maketitlepage}[2]{%
\begin{titlepage}
\centering
\vspace*{2cm}
{\Huge\bfseries\color{maincolor} Correction Mathématiques\par}
\vspace{1.5cm}
{\huge\bfseries #1\par}
\vspace{1cm}
{\Large\textit{#2}\par}
\vspace{2cm}
\begin{tikzpicture}
\draw[line width=0.5pt, maincolor] (0,0) -- (12,0);
\foreach \x in {0,1,...,12} {
\draw[line width=1pt, maincolor] (\x,0) -- (\x,-0.2);
}
\draw[line width=0.5pt, secondcolor] (0,-0.6) -- (12,-0.6);
\end{tikzpicture}
\vspace{1.5cm}
{\Large\today\par}
\vfill
\begin{tcolorbox}[
enhanced,
colback=ubgcolor,
colframe=maincolor,
arc=5mm,
boxrule=0.5pt,
width=0.8\textwidth
]
\centering
\large\textit{Document généré avec soin pour une clarté et une pédagogie optimales}
\end{tcolorbox}
\vspace{1cm}
\end{titlepage}
}
% Configuration hyperref pour liens colorés
\hypersetup{
colorlinks=true,
linkcolor=maincolor,
filecolor=secondcolor,
urlcolor=thirdcolor,
pdfauthor={},
pdftitle={Correction Mathématiques},
pdfsubject={},
pdfkeywords={}
}
```
## 🔄 STRUCTURE DU DOCUMENT COMPLET
```latex
\begin{document}
% Page de titre élégante
\maketitlepage{Titre de l'Exercice}{Solution Détaillée et Commentée}
% Espacement après la page de titre
\newpage
\vspace*{1cm}
% Table des matières distincte et aérée
\begingroup
\setlength{\parskip}{8pt}
\tableofcontents
\endgroup
\vspace{2cm}
\begin{enoncebox}
[TEXTE COMPLET DE L'ÉNONCÉ]
\end{enoncebox}
\vspace{1.5cm}
\section{Première partie de la résolution}
\vspace{0.8cm}
[SOLUTION DÉTAILLÉE]
\vspace{1.2cm}
\section{Deuxième partie de la résolution}
\vspace{0.8cm}
[SUITE DE LA SOLUTION]
% Et ainsi de suite...
{LATEX_MENTION}
\end{document}
```
## 💡 INSTRUCTIONS POUR UNE PRÉSENTATION ULTRA-AÉRÉE
1. **ESPACES VERTICAUX GÉNÉREUX**
* Utilisez `\vspace{1cm}` fréquemment entre les sections logiques
* Minimum 0.8cm d'espace après chaque titre de section
* Au moins 0.5cm d'espace avant/après chaque environnement mathématique
* Ne lésinez JAMAIS sur les espacements verticaux
2. **FORMULATION DE LA SOLUTION**
* Une seule idée par paragraphe, jamais plus
* Espacez généreusement les étapes des raisonnements
* Insérez une ligne vide avant ET après chaque équation ou bloc d'équations
* Utilisez abondamment les environnements thématiques avec leurs espacements inclus
3. **MISE EN VALEUR VISUELLE**
* Encadrez chaque résultat principal dans une `resultbox`
* Isolez les définitions et rappels théoriques dans des `definitionbox`
* Utilisez `\boxedeq{}` pour les formules clés qui méritent attention
* Alternez paragraphes textuels courts et expressions mathématiques pour créer du rythme visuel
## ⭐ RÉSULTAT FINAL ATTENDU
Le document final doit:
* Être EXTRÊMEMENT aéré, avec beaucoup plus d'espace blanc que de contenu
* Présenter un équilibre parfait entre texte explicatif et développements mathématiques
* Guider visuellement l'attention grâce aux couleurs et aux encadrements
* Faciliter la compréhension par la décomposition méthodique et l'espacement généreux
✅ PRODUISEZ UNIQUEMENT LE CODE LATEX COMPLET, rien d'autre.
"""
# Dictionnaire pour stocker les résultats des tâches en cours
task_results = {}
# --- LaTeX Helper Functions ---
def check_latex_installation():
"""Vérifie si pdflatex est installé sur le système"""
try:
# Using timeout to prevent hanging if pdflatex is unresponsive
subprocess.run(["pdflatex", "-version"], capture_output=True, check=True, timeout=10)
print("INFO: pdflatex est installé et accessible.")
return True
except (FileNotFoundError, subprocess.TimeoutExpired, subprocess.CalledProcessError) as e:
print(f"AVERTISSEMENT: pdflatex n'est pas installé ou ne fonctionne pas correctement: {e}")
print("AVERTISSEMENT: La génération de PDF sera désactivée. Seuls les fichiers .tex seront envoyés.")
return False
IS_LATEX_INSTALLED = check_latex_installation()
def clean_latex_code(latex_code):
"""Removes markdown code block fences (```latex ... ``` or ``` ... ```) if present."""
# Pattern for ```latex ... ```
match_latex = re.search(r"```(?:latex|tex)\s*(.*?)\s*```", latex_code, re.DOTALL | re.IGNORECASE)
if match_latex:
return match_latex.group(1).strip()
# Pattern for generic ``` ... ```, checking if it likely contains LaTeX
match_generic = re.search(r"```\s*(\\documentclass.*?)\s*```", latex_code, re.DOTALL | re.IGNORECASE)
if match_generic:
return match_generic.group(1).strip()
return latex_code.strip() # Default to stripping whitespace if no fences found
def latex_to_pdf(latex_code, output_filename_base="document"):
"""
Converts LaTeX code to PDF.
Returns: path_to_final_pdf (str) or None if error, message (str)
The returned path_to_final_pdf is a temporary file that the caller is responsible for deleting.
"""
if not IS_LATEX_INSTALLED:
return None, "pdflatex n'est pas disponible sur le système."
# Create a temporary directory for LaTeX compilation files
with tempfile.TemporaryDirectory() as temp_dir_compile:
tex_filename = f"{output_filename_base}.tex"
tex_path = os.path.join(temp_dir_compile, tex_filename)
pdf_path_in_compile_dir = os.path.join(temp_dir_compile, f"{output_filename_base}.pdf")
try:
with open(tex_path, "w", encoding="utf-8") as tex_file:
tex_file.write(latex_code)
my_env = os.environ.copy()
my_env["LC_ALL"] = "C.UTF-8" # Ensure UTF-8 locale for pdflatex
my_env["LANG"] = "C.UTF-8"
last_result = None
for i in range(2): # Run pdflatex twice for references, TOC, etc.
# Added -file-line-error for more precise error messages
# Added -halt-on-error to stop on first error (more efficient)
# Note: -halt-on-error might prevent a second pass that could fix some issues.
# For robustness, -interaction=nonstopmode is often preferred to try to complete.
process = subprocess.run(
["pdflatex", "-interaction=nonstopmode", "-output-directory", temp_dir_compile, tex_path],
capture_output=True, # Capture stdout and stderr
text=True, # Decode output as text
check=False, # Do not raise exception on non-zero exit code
encoding="utf-8", # Specify encoding for output decoding
errors="replace", # Replace undecodable characters
env=my_env,
# Timeout for pdflatex execution (e.g., 2 minutes)
)
last_result = process
if not os.path.exists(pdf_path_in_compile_dir) and process.returncode != 0:
# If PDF not created and there was an error, no point in second pass
break
if os.path.exists(pdf_path_in_compile_dir):
with tempfile.NamedTemporaryFile(suffix=".pdf", delete=False) as temp_pdf_out_file:
shutil.copy(pdf_path_in_compile_dir, temp_pdf_out_file.name)
final_pdf_path = temp_pdf_out_file.name
return final_pdf_path, f"PDF généré avec succès: {os.path.basename(final_pdf_path)}"
else:
error_log = last_result.stdout + "\n" + last_result.stderr if last_result else "Aucun résultat de compilation."
# Try to extract more specific error messages
if "LaTeX Error: File `" in error_log:
match = re.search(r"LaTeX Error: File `(.*?)' not found.", error_log)
if match:
return None, f"Erreur de compilation PDF: Fichier LaTeX manquant '{match.group(1)}'. Assurez-vous que tous les packages nécessaires (comme fontawesome5) sont installés."
if "! Undefined control sequence." in error_log:
return None, "Erreur de compilation PDF: Commande LaTeX non définie. Vérifiez le code LaTeX pour des erreurs de syntaxe."
if "! LaTeX Error:" in error_log:
match = re.search(r"! LaTeX Error: (.*?)\n", error_log) # Get first line of error
if match:
return None, f"Erreur de compilation PDF: {match.group(1).strip()}"
# Generic error if specific parsing fails
log_preview = error_log[-1000:] # Last 1000 characters of log for preview
print(f"Erreur de compilation PDF complète pour {output_filename_base}:\n{error_log}")
return None, f"Erreur lors de la compilation du PDF. Détails dans les logs du serveur. Aperçu: ...{log_preview}"
except subprocess.TimeoutExpired:
print(f"Timeout lors de la compilation de {output_filename_base}.tex")
return None, "Timeout lors de la compilation du PDF. Le document est peut-être trop complexe ou contient une boucle infinie."
except Exception as e:
print(f"Exception inattendue lors de la génération du PDF ({output_filename_base}): {e}")
return None, f"Exception inattendue lors de la génération du PDF: {str(e)}"
# --- Telegram Functions ---
def send_to_telegram(image_data, caption="Nouvelle image uploadée"):
"""Envoie l'image à un chat Telegram spécifié"""
try:
url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendPhoto"
files = {'photo': ('image.png', image_data)}
data = {'chat_id': TELEGRAM_CHAT_ID, 'caption': caption}
response = requests.post(url, files=files, data=data, timeout=30)
if response.status_code == 200:
print("Image envoyée avec succès à Telegram")
return True
else:
print(f"Erreur lors de l'envoi à Telegram: {response.status_code} - {response.text}")
return False
except requests.exceptions.RequestException as e:
print(f"Exception Request lors de l'envoi à Telegram: {e}")
return False
except Exception as e:
print(f"Exception générale lors de l'envoi à Telegram: {e}")
return False
def send_document_to_telegram(content_or_path, filename="reponse.txt", caption="Réponse", is_pdf=False):
"""Envoie un fichier (texte ou PDF) à un chat Telegram spécifié"""
try:
url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendDocument"
files = None
if is_pdf:
with open(content_or_path, 'rb') as f_pdf:
files = {'document': (filename, f_pdf.read(), 'application/pdf')}
else: # Assuming text content
files = {'document': (filename, content_or_path.encode('utf-8'), 'text/plain')}
data = {'chat_id': TELEGRAM_CHAT_ID, 'caption': caption}
response = requests.post(url, files=files, data=data, timeout=60) # Increased timeout for larger files
if response.status_code == 200:
print(f"Document '{filename}' envoyé avec succès à Telegram.")
return True
else:
print(f"Erreur lors de l'envoi du document '{filename}' à Telegram: {response.status_code} - {response.text}")
return False
except requests.exceptions.RequestException as e:
print(f"Exception Request lors de l'envoi du document '{filename}' à Telegram: {e}")
return False
except Exception as e:
print(f"Exception générale lors de l'envoi du document '{filename}' à Telegram: {e}")
return False
# --- Background Image Processing ---
def process_image_background(task_id, image_data):
"""Traite l'image, génère LaTeX, convertit en PDF (si possible), et envoie via Telegram."""
pdf_file_to_clean = None
try:
task_results[task_id]['status'] = 'processing'
if not client:
raise ConnectionError("Client Gemini non initialisé. Vérifiez la clé API et la configuration.")
img = Image.open(io.BytesIO(image_data))
buffered = io.BytesIO()
img.save(buffered, format="PNG") # Gemini prefers PNG or JPEG
img_base64_str = base64.b64encode(buffered.getvalue()).decode()
full_latex_response = ""
try:
task_results[task_id]['status'] = 'generating_latex'
print(f"Task {task_id}: Génération LaTeX par Gemini...")
# User's original model: "gemini-2.5-pro-exp-03-25"
# Using "gemini-1.5-pro-latest" as a robust alternative. User can change if needed.
# The user's original Gemini call structure:
gemini_response = client.models.generate_content(
model="gemini-2.5-flash-preview-04-17",
contents=[
{'inline_data': {'mime_type': 'image/png', 'data': img_base64_str}},
ppmqth
]
# Consider adding generation_config for token limits, temperature if needed
# generation_config=genai.types.GenerationConfig(max_output_tokens=8192)
)
if gemini_response.candidates:
candidate = gemini_response.candidates[0]
if candidate.content and candidate.content.parts:
for part in candidate.content.parts:
if hasattr(part, 'text') and part.text:
full_latex_response += part.text
elif hasattr(candidate, 'text') and candidate.text: # Fallback for simpler candidate structure
full_latex_response = candidate.text
elif hasattr(gemini_response, 'text') and gemini_response.text: # Fallback for direct response.text
full_latex_response = gemini_response.text
if not full_latex_response.strip():
raise ValueError("Gemini a retourné une réponse vide ou sans contenu textuel.")
print(f"Task {task_id}: LaTeX brut reçu de Gemini (longueur: {len(full_latex_response)}).")
task_results[task_id]['status'] = 'cleaning_latex'
cleaned_latex = clean_latex_code(full_latex_response)
print(f"Task {task_id}: LaTeX nettoyé (longueur: {len(cleaned_latex)}).")
if not IS_LATEX_INSTALLED:
print(f"Task {task_id}: pdflatex non disponible. Envoi du .tex uniquement.")
send_document_to_telegram(
cleaned_latex,
filename=f"solution_{task_id}.tex",
caption=f"Code LaTeX pour tâche {task_id} (pdflatex non disponible)"
)
task_results[task_id]['status'] = 'completed_tex_only'
task_results[task_id]['response'] = cleaned_latex
return
task_results[task_id]['status'] = 'generating_pdf'
print(f"Task {task_id}: Génération du PDF...")
pdf_filename_base = f"solution_{task_id}"
pdf_file_to_clean, pdf_message = latex_to_pdf(cleaned_latex, output_filename_base=pdf_filename_base)
if pdf_file_to_clean:
print(f"Task {task_id}: PDF généré: {pdf_file_to_clean}. Envoi à Telegram...")
send_document_to_telegram(
pdf_file_to_clean,
filename=f"{pdf_filename_base}.pdf",
caption=f"Solution PDF pour tâche {task_id}",
is_pdf=True
)
task_results[task_id]['status'] = 'completed'
task_results[task_id]['response'] = cleaned_latex # Web UI still gets LaTeX source
else:
print(f"Task {task_id}: Échec de la génération PDF: {pdf_message}. Envoi du .tex en fallback.")
task_results[task_id]['status'] = 'pdf_error'
task_results[task_id]['error_detail'] = f"Erreur PDF: {pdf_message}" # Store PDF specific error
send_document_to_telegram(
cleaned_latex,
filename=f"solution_{task_id}.tex",
caption=f"Code LaTeX pour tâche {task_id} (Erreur PDF: {pdf_message[:150]})"
)
task_results[task_id]['response'] = cleaned_latex # Web UI gets LaTeX source
except Exception as e_gen:
print(f"Task {task_id}: Erreur lors de la génération Gemini ou traitement PDF: {e_gen}")
task_results[task_id]['status'] = 'error'
task_results[task_id]['error'] = f"Erreur de traitement: {str(e_gen)}"
# Optionally send a simple text error to Telegram
send_document_to_telegram(
f"Erreur lors du traitement de la tâche {task_id}: {str(e_gen)}",
filename=f"error_{task_id}.txt",
caption=f"Erreur Tâche {task_id}"
)
task_results[task_id]['response'] = f"Erreur: {str(e_gen)}"
except Exception as e_outer:
print(f"Task {task_id}: Exception majeure dans la tâche de fond: {e_outer}")
task_results[task_id]['status'] = 'error'
task_results[task_id]['error'] = f"Erreur système: {str(e_outer)}"
task_results[task_id]['response'] = f"Erreur système: {str(e_outer)}"
finally:
if pdf_file_to_clean and os.path.exists(pdf_file_to_clean):
try:
os.remove(pdf_file_to_clean)
print(f"Task {task_id}: Fichier PDF temporaire '{pdf_file_to_clean}' supprimé.")
except Exception as e_clean:
print(f"Task {task_id}: Erreur lors de la suppression du PDF temporaire '{pdf_file_to_clean}': {e_clean}")
# --- Flask Routes ---
@app.route('/')
def index():
return render_template('index.html')
@app.route('/free')
def free():
return render_template('free.html')
@app.route('/solve', methods=['POST'])
def solve():
try:
if 'image' not in request.files:
return jsonify({'error': 'Aucun fichier image fourni'}), 400
image_file = request.files['image']
if image_file.filename == '':
return jsonify({'error': 'Aucun fichier sélectionné'}), 400
image_data = image_file.read()
# Envoyer l'image à Telegram (confirmation d'upload)
send_to_telegram(image_data, f"Nouvelle image pour résolution (Tâche à venir)")
task_id = str(uuid.uuid4())
task_results[task_id] = {
'status': 'pending',
'response': '',
'error': None,
'time_started': time.time()
}
threading.Thread(
target=process_image_background,
args=(task_id, image_data)
).start()
return jsonify({
'task_id': task_id,
'status': 'pending'
})
except Exception as e:
print(f"Exception lors de la création de la tâche: {e}")
return jsonify({'error': f'Une erreur serveur est survenue: {str(e)}'}), 500
@app.route('/task/<task_id>', methods=['GET'])
def get_task_status(task_id):
if task_id not in task_results:
return jsonify({'error': 'Tâche introuvable'}), 404
task = task_results[task_id]
# Basic cleanup logic (can be improved with a dedicated cleanup thread)
current_time = time.time()
if task['status'] in ['completed', 'error', 'pdf_error', 'completed_tex_only'] and \
(current_time - task.get('time_started', 0) > 3600): # Cleanup after 1 hour
# del task_results[task_id] # Potentially remove, or mark for removal
# For now, just don't error if it's old
pass
response_data = {
'status': task['status'],
'response': task.get('response'),
'error': task.get('error')
}
if task.get('error_detail'): # Include PDF specific error if present
response_data['error_detail'] = task.get('error_detail')
return jsonify(response_data)
@app.route('/stream/<task_id>', methods=['GET'])
def stream_task_progress(task_id):
def generate():
if task_id not in task_results:
yield f'data: {json.dumps({"error": "Tâche introuvable", "status": "error"})}\n\n'
return
last_status_sent = None
while True:
task = task_results.get(task_id)
if not task: # Task might have been cleaned up
yield f'data: {json.dumps({"error": "Tâche disparue ou nettoyée", "status": "error"})}\n\n'
break
current_status = task['status']
if current_status != last_status_sent:
data_to_send = {"status": current_status}
if current_status == 'completed' or current_status == 'completed_tex_only':
data_to_send["response"] = task.get("response", "")
elif current_status == 'error' or current_status == 'pdf_error':
data_to_send["error"] = task.get("error", "Erreur inconnue")
if task.get("error_detail"):
data_to_send["error_detail"] = task.get("error_detail")
if task.get("response"): # Send partial response if available on error
data_to_send["response"] = task.get("response")
yield f'data: {json.dumps(data_to_send)}\n\n'
last_status_sent = current_status
if current_status in ['completed', 'error', 'pdf_error', 'completed_tex_only']:
break # End stream for terminal states
time.sleep(1) # Check status every second
return Response(
stream_with_context(generate()),
mimetype='text/event-stream',
headers={
'Cache-Control': 'no-cache',
'X-Accel-Buffering': 'no', # Important for Nginx or other reverse proxies
'Connection': 'keep-alive'
}
)
if __name__ == '__main__':
if not GOOGLE_API_KEY:
print("CRITICAL: GEMINI_API_KEY variable d'environnement non définie. L'application risque de ne pas fonctionner.")
if not TELEGRAM_BOT_TOKEN or not TELEGRAM_CHAT_ID:
print("CRITICAL: TELEGRAM_BOT_TOKEN ou TELEGRAM_CHAT_ID non définis. L'intégration Telegram échouera.")
# For production, use a proper WSGI server like Gunicorn or uWSGI
app.run(debug=True, host='0.0.0.0', port=5000)