Update app.py

app.py

@@ -1,65 +1,82 @@
 import gradio as gr
-from equity_assistant import EquityAssistant, DesignRecommender, InteractionManager, LearningModule
 
-class EquitySpace:
+class EquityAssistant:
     def __init__(self):
-        self.assistant = EquityAssistant()
-        self.recommender = DesignRecommender()
-        self.interaction = InteractionManager()
-        self.learning = LearningModule()
+        self.user_preferences = {
+            'style_history': [],
+            'color_preferences': [],
+            'layout_choices': [],
+            'expertise_level': 'novice'
+        }
 
-    def process_consultation(self, poster_type, style_preferences, main_message):
-        """Traite la consultation initiale et retourne des recommandations"""
-        # Mise à jour des préférences utilisateur
-        self.assistant.user_preferences['style_history'].append(style_preferences)
+    def process_request(self, poster_type, style_pref, message):
+        """Traite la demande de l'utilisateur et génère des recommandations"""
+        recommendations = {
+            'layout': self.suggest_layout(poster_type),
+            'style': self.process_style(style_pref),
+            'message': self.analyze_message(message)
+        }
         
-        # Génération des recommandations
-        color_scheme = self.recommender.recommend_colors(poster_type, {})
-        layout = self.recommender.suggest_layout(poster_type, main_message)
-        typography = self.recommender.recommend_typography(style_preferences, main_message)
-        
-        response = f"""
-        Recommandations pour votre affiche :
-        
-        Type d'affiche: {poster_type}
-        Style choisi: {style_preferences}
-        Message principal: {main_message}
-        
-        Suggestions:
-        - Mise en page: {layout if layout else 'Layout suggestions will appear here'}
-        - Typographie: {typography if typography else 'Typography suggestions will appear here'}
-        - Couleurs: {color_scheme if color_scheme else 'Color scheme suggestions will appear here'}
+        return self.format_response(recommendations)
+    
+    def suggest_layout(self, poster_type):
+        layouts = {
+            "événement": "Layout vertical avec focus sur la date et le lieu",
+            "présentation": "Layout structuré avec hiérarchie claire",
+            "infographie": "Layout en grille ou flux avec données visuelles",
+            "autre": "Layout flexible selon le contenu"
+        }
+        return layouts.get(poster_type.lower(), "Layout personnalisé selon vos besoins")
+    
+    def process_style(self, style_pref):
+        return f"Style recommandé basé sur vos préférences : {style_pref}"
+    
+    def analyze_message(self, message):
+        return f"Recommendations pour votre message : {message}"
+    
+    def format_response(self, recommendations):
+        return f"""
+📋 Recommandations pour votre affiche :
+
+🎯 Layout suggéré :
+{recommendations['layout']}
+
+🎨 Style :
+{recommendations['style']}
+
+💡 Message :
+{recommendations['message']}
+
+Tips pour l'équité :
+• Utilisez des contrastes adaptés pour la lisibilité
+• Privilégiez des polices inclusives
+• Vérifiez l'accessibilité des couleurs
         """
-        
-        return response
 
-# Configuration de l'interface Gradio
-def create_interface():
-    equity_space = EquitySpace()
-    
-    # Création de l'interface
-    iface = gr.Interface(
-        fn=equity_space.process_consultation,
-        inputs=[
-            gr.Dropdown(
-                choices=["Affiche événement", "Présentation", "Infographie", "Autre"],
-                label="Type d'affiche"
-            ),
-            gr.Textbox(label="Préférences de style (ex: moderne, minimaliste, coloré...)"),
-            gr.Textbox(label="Message principal à communiquer")
-        ],
-        outputs=gr.Textbox(label="Recommandations"),
-        title="Assistant Equity - Création d'affiches",
-        description="""
-        Bienvenue dans l'Assistant Equity! 
-        Cet outil vous aide à créer des affiches efficaces et équitables.
-        Commencez par répondre aux questions ci-dessous pour recevoir des recommandations personnalisées.
-        """,
-        theme="huggingface"
-    )
-    return iface
+# Création de l'interface
+assistant = EquityAssistant()
+
+demo = gr.Interface(
+    fn=assistant.process_request,
+    inputs=[
+        gr.Dropdown(
+            choices=["Événement", "Présentation", "Infographie", "Autre"],
+            label="Type d'affiche"
+        ),
+        gr.Textbox(
+            label="Style souhaité",
+            placeholder="Ex: moderne, minimaliste, coloré..."
+        ),
+        gr.Textbox(
+            label="Message principal",
+            placeholder="Décrivez le message clé de votre affiche"
+        )
+    ],
+    outputs=gr.Textbox(label="Recommandations"),
+    title="🎨 Equity Assistant",
+    description="Assistant IA pour la création d'affiches équitables et efficaces",
+    theme="soft"
+)
 
-# Lancement de l'application
 if __name__ == "__main__":
-    interface = create_interface()
-    interface.launch()
+    demo.launch()