Create app.py

app.py

@@ -0,0 +1,91 @@
+import gradio as gr
+import requests
+import base64
+import io
+from PIL import Image
+
+# METTEZ VOTRE CLÉ API MISTRAL ICI (créée sur console.mistral.ai)
+MISTRAL_API_KEY = "b9rg7dcD7IBHyNiUFVI5wa57zBSjsPKx"  # ← Remplacez par votre clé
+API_URL = "https://api.mistral.ai/v1/chat/completions"
+
+def encode_image(image):
+    """Convertit l'image en base64"""
+    buffer = io.BytesIO()
+    image.save(buffer, format="JPEG")
+    return base64.b64encode(buffer.getvalue()).decode()
+
+def generate_filename(image, custom_prompt):
+    """Génère un nom de fichier structuré"""
+    
+    if image is None:
+        return "❌ Veuillez uploader une image."
+    
+    if not MISTRAL_API_KEY or MISTRAL_API_KEY == "VOTRE_CLE_API_MISTRAL":
+        return "❌ Clé API manquante. Obtenez une clé gratuite sur console.mistral.ai"
+    
+    try:
+        image_base64 = encode_image(image)
+        
+        if custom_prompt and custom_prompt.strip():
+            prompt = custom_prompt.strip()
+        else:
+            prompt = """Analyse ce document. Extrait le nom des parties et la date.
+Génère un nom de fichier au format: Affaire_[X]_contre_[Y]_[JJ_MM_AAAA]
+Exemple: Affaire_Dupont_contre_Martin_12_12_2026
+Réponds uniquement par le nom de fichier, rien d'autre."""
+        
+        response = requests.post(
+            API_URL,
+            headers={
+                "Authorization": f"Bearer {MISTRAL_API_KEY}",
+                "Content-Type": "application/json"
+            },
+            json={
+                "model": "pixtral-12b-2409",
+                "messages": [{
+                    "role": "user",
+                    "content": [
+                        {"type": "text", "text": prompt},
+                        {"type": "image_url", "image_url": f"data:image/jpeg;base64,{image_base64}"}
+                    ]
+                }],
+                "max_tokens": 100,
+                "temperature": 0.1
+            },
+            timeout=60
+        )
+        
+        if response.status_code == 200:
+            result = response.json()
+            return result["choices"][0]["message"]["content"].strip()
+        else:
+            return f"❌ Erreur: {response.status_code}"
+            
+    except Exception as e:
+        return f"❌ Erreur: {str(e)}"
+
+# Interface
+with gr.Blocks(title="Nommage de documents juridiques") as demo:
+    gr.Markdown("""
+    # ⚖️ Générateur de noms pour documents juridiques
+    **Basé sur Mistral Pixtral** - Gratuit
+    
+    Upload un document (photo, scan) → obtient `Affaire_X_contre_Y_DATE`
+    """)
+    
+    with gr.Row():
+        with gr.Column():
+            image = gr.Image(type="pil", label="Document", height=400)
+            prompt = gr.Textbox(
+                label="Format personnalisé (optionnel)",
+                placeholder='Ex: "Format: DOS_[X]_VS_[Y]_[DDMMYYYY]"',
+                lines=2
+            )
+            btn = gr.Button("Générer le nom", variant="primary")
+        
+        with gr.Column():
+            output = gr.Textbox(label="Nom généré", lines=5)
+    
+    btn.click(fn=generate_filename, inputs=[image, prompt], outputs=output)
+
+demo.launch()