Update app.py

app.py

@@ -2,95 +2,75 @@ import gradio as gr
 
 class PromptExpert:
     def __init__(self):
-        self.templates = {
-            "expert": {
-                "structure": """Description technique détaillée:
-{prompt}
-
-Paramètres techniques:
-- Résolution: {quality}%
-- Éclairage: précis et équilibré 
-- Niveau de détail: maximum
-- Format: ultra-haute définition
-
-Exigences:
-1. Composition professionnelle
-2. Netteté exceptionnelle 
-3. Rendu photoréaliste
-4. Respect des standards professionnels""",
-                "focus": ["Qualité technique", "Précision", "Standards pro"]
-            },
-            "creative": {
-                "structure": """Vision créative:
-{prompt}
-
-Style artistique:
-- Innovation: {quality}%
-- Approche: expérimentale
-- Expression: libre
-- Ambiance: {mood}
-
-Éléments créatifs:
-1. {creative_element1}
-2. {creative_element2}
-3. {creative_element3}
-
-Focus artistique:
-1. Composition unique
-2. Effets visuels innovants
-3. Impact émotionnel""",
-                "moods": ["dramatique", "onirique", "dynamique", "serein"],
-                "elements": [
-                    "Jeu de lumières spectaculaire",
-                    "Perspective originale",
-                    "Textures complexes",
-                    "Mouvement fluide",
-                    "Atmosphère immersive",
-                    "Contraste saisissant"
-                ]
-            }
+        self.expert_intros = {
+            "étude": "En tant qu'analyste stratégique spécialisé dans",
+            "marché": "En qualité d'expert en études de marché focalisé sur",
+            "image": "En tant que directeur artistique spécialisé dans",
+            "technique": "En tant qu'expert technique dans le domaine de",
+            "recherche": "En qualité de chercheur spécialisé dans",
+            "développement": "En tant que développeur expert en",
+            "business": "En qualité de consultant stratégique spécialisé en",
+            "default": "En tant qu'expert dans le domaine de"
         }
 
-    def _get_creative_elements(self, base_prompt):
-        import random
-        elements = self.templates["creative"]["elements"]
-        chosen = random.sample(elements, 3)
-        return chosen
+    def _detect_domain(self, prompt):
+        prompt_lower = prompt.lower()
+        for key in self.expert_intros.keys():
+            if key in prompt_lower:
+                return key
+        return "default"
 
-    def _get_mood(self, base_prompt):
-        import random
-        return random.choice(self.templates["creative"]["moods"])
+    def _add_expert_context(self, domain, topic):
+        context_templates = {
+            "étude": f"{topic}, je souhaite mener une analyse approfondie couvrant :\n\n" +
+                    "1. L'état actuel du marché et ses tendances\n" +
+                    "2. Les opportunités et menaces identifiées\n" +
+                    "3. Les facteurs clés de succès\n" +
+                    "4. Les recommandations stratégiques\n\n" +
+                    "L'analyse devra être basée sur des données quantitatives et qualitatives récentes.",
+            
+            "image": f"{topic}, avec les caractéristiques suivantes :\n\n" +
+                    "1. Style visuel ultra-détaillé et photoréaliste\n" +
+                    "2. Composition dynamique et équilibrée\n" +
+                    "3. Éclairage cinématographique\n" +
+                    "4. Rendu en ultra-haute définition\n\n" +
+                    "Le résultat doit captiver visuellement et transmettre une émotion forte.",
+            
+            "default": f"{topic}, en mettant l'accent sur :\n\n" +
+                    "1. Une approche méthodologique rigoureuse\n" +
+                    "2. Des données et arguments solides\n" +
+                    "3. Des conclusions actionnables\n" +
+                    "4. Des recommandations concrètes\n\n" +
+                    "L'analyse doit être exhaustive et orientée résultats."
+        }
+        return context_templates.get(domain, context_templates["default"])
 
     def optimize(self, prompt, mode, complexity):
-        if mode == "creative":
-            template = self.templates[mode]["structure"]
-            mood = self._get_mood(prompt)
-            elements = self._get_creative_elements(prompt)
-            return template.format(
-                prompt=prompt,
-                quality=complexity,
-                mood=mood,
-                creative_element1=elements[0],
-                creative_element2=elements[1],
-                creative_element3=elements[2]
-            )
+        domain = self._detect_domain(prompt)
+        intro = self.expert_intros[domain]
+        expert_context = self._add_expert_context(domain, prompt)
+        
+        if mode == "expert":
+            return f"{intro} {expert_context}\n\n" + \
+                   f"Niveau de détail: {complexity}%\n" + \
+                   "Objectif: Produire une analyse exhaustive et actionnable."
         else:
-            return self.templates["expert"]["structure"].format(
-                prompt=prompt,
-                quality=complexity
-            )
+            return f"Approche créative pour {prompt}\n\n" + \
+                   f"Innovation: {complexity}%\n" + \
+                   "Focus: Explorer des angles nouveaux et non conventionnels."
 
 def create_interface():
     expert = PromptExpert()
 
-    with gr.Blocks(title="Optimiseur de Prompts IA") as demo:
+    with gr.Blocks() as demo:
         gr.Markdown("# Optimiseur de Prompts IA")
         
         with gr.Row():
             with gr.Column():
                 input_prompt = gr.Textbox(
                     label="Prompt Initial",
-                    placeholder="Décrivez votre image..."
+                    placeholder="Décrivez votre demande...",
+                    lines=3
                 )
                 
                 mode = gr.Radio(
@@ -102,8 +82,8 @@ def create_interface():
                 complexity = gr.Slider(
                     minimum=0,
                     maximum=100,
-                    value=75,
-                    label="Niveau de Complexité"
+                    value=85,
+                    label="Niveau de Détail"
                 )
                 
                 optimize_btn = gr.Button("Optimiser")