Update app.py

app.py

@@ -6,9 +6,8 @@ from typing import Dict, List
 
 class PromptGenerator:
     def __init__(self):
-        # Carrega o modelo Mistral em português do Hugging Face
         print("Carregando modelo... (pode demorar alguns minutos na primeira vez)")
-        self.model_name = "beomi/llama-2-ko-7b"  # Modelo base que funciona bem com português
+        self.model_name = "beomi/llama-2-ko-7b"
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
         self.model = AutoModelForCausalLM.from_pretrained(
             self.model_name,
@@ -18,9 +17,7 @@ class PromptGenerator:
         print("Modelo carregado!")
 
     def generate_with_model(self, prompt: str) -> str:
-        """Gera texto usando o modelo local"""
         inputs = self.tokenizer(prompt, return_tensors="pt").to(self.model.device)
-        
         outputs = self.model.generate(
             **inputs,
             max_new_tokens=512,
@@ -29,10 +26,8 @@ class PromptGenerator:
             do_sample=True,
             pad_token_id=self.tokenizer.eos_token_id
         )
-        
         return self.tokenizer.decode(outputs[0], skip_special_tokens=True)
 
-# Templates de prompts por categoria
 TEMPLATES = {
     "história": {
         "creative": "Crie uma história {gênero} sobre {tema} com {personagens} personagens. A história deve ter {comprimento} palavras e incluir {elementos}.",
@@ -51,20 +46,16 @@ TEMPLATES = {
     }
 }
 
-# Inicializa o gerador de prompts
 generator = PromptGenerator()
 
 def generate_prompt(categoria: str, subcategoria: str, tema: str, elementos: str, 
                    público_alvo: str, tom: str, comprimento: int, 
                    usar_llm: bool = True) -> str:
-    """Gera um prompt baseado nos parâmetros e usa o modelo para expandir"""
-    
     if categoria not in TEMPLATES or subcategoria not in TEMPLATES[categoria]:
         return "Categoria ou subcategoria não encontrada"
     
     template = TEMPLATES[categoria][subcategoria]
     
-    # Parâmetros base para o template
     params = {
         "tema": tema,
         "elementos": elementos,
@@ -82,14 +73,12 @@ def generate_prompt(categoria: str, subcategoria: str, tema: str, elementos: str
         "analogias": "analogias cotidianas"
     }
     
-    # Gera o prompt base
     base_prompt = template.format(**params)
     base_prompt += f"\n\nTom desejado: {tom}"
     
     if comprimento:
         base_prompt += f"\nComprimento aproximado: {comprimento} palavras"
     
-    # Se solicitado, usa o modelo para expandir o prompt
     if usar_llm:
         instruction = f"""
         Você é um assistente especializado em criar prompts detalhados em português para LLMs.
@@ -105,58 +94,111 @@ def generate_prompt(categoria: str, subcategoria: str, tema: str, elementos: str
     return base_prompt
 
 def create_interface():
-    """Cria a interface do Gradio"""
-    
-    with gr.Blocks(title="Gerador de Prompts para LLMs") as app:
-        gr.Markdown("# 🤖 Gerador de Prompts Inteligente")
-        gr.Markdown("Crie prompts estruturados e detalhados usando IA")
+    with gr.Blocks(
+        title="Gerador de Prompts para LLMs",
+        theme=gr.themes.Soft(
+            primary_hue="indigo",
+            secondary_hue="blue",
+        ),
+    ) as app:
+        gr.Markdown(
+            """
+            # 🤖 Gerador de Prompts Inteligente
+            ### Crie prompts estruturados e detalhados usando IA
+            """
+        )
         
-        with gr.Row():
-            with gr.Column():
-                categoria = gr.Dropdown(
-                    choices=list(TEMPLATES.keys()),
-                    label="Categoria",
-                    value="história"
-                )
-                subcategoria = gr.Dropdown(
-                    choices=list(TEMPLATES["história"].keys()),
-                    label="Subcategoria",
-                    value="creative"
-                )
-                tema = gr.Textbox(
-                    label="Tema Principal",
-                    placeholder="Ex: inteligência artificial, sustentabilidade"
-                )
+        with gr.Tabs():
+            with gr.TabItem("✨ Gerador de Prompts"):
+                with gr.Group():
+                    with gr.Row():
+                        with gr.Column(scale=1):
+                            categoria = gr.Dropdown(
+                                choices=list(TEMPLATES.keys()),
+                                label="📚 Categoria",
+                                value="história",
+                                container=False,
+                            )
+                            subcategoria = gr.Dropdown(
+                                choices=list(TEMPLATES["história"].keys()),
+                                label="🔍 Subcategoria",
+                                value="creative",
+                                container=False,
+                            )
+                        
+                        with gr.Column(scale=2):
+                            tema = gr.Textbox(
+                                label="💡 Tema Principal",
+                                placeholder="Ex: inteligência artificial, sustentabilidade",
+                                container=False,
+                            )
+                            elementos = gr.Textbox(
+                                label="🔮 Elementos Específicos",
+                                placeholder="Ex: conflitos, conceitos, exemplos práticos",
+                                container=False,
+                            )
+                    
+                    with gr.Row():
+                        with gr.Column(scale=1):
+                            público_alvo = gr.Textbox(
+                                label="👥 Público Alvo",
+                                placeholder="Ex: iniciantes, profissionais, estudantes",
+                                container=False,
+                            )
+                            tom = gr.Dropdown(
+                                choices=["formal", "informal", "técnico", "conversacional", "educativo"],
+                                label="🎭 Tom",
+                                value="formal",
+                                container=False,
+                            )
+                        
+                        with gr.Column(scale=2):
+                            comprimento = gr.Slider(
+                                minimum=100,
+                                maximum=2000,
+                                step=100,
+                                label="📏 Comprimento (palavras)",
+                                value=500,
+                                container=False,
+                            )
+                            usar_llm = gr.Checkbox(
+                                label="🤖 Usar IA para expandir prompt",
+                                value=True,
+                                container=False,
+                            )
                 
-            with gr.Column():
-                elementos = gr.Textbox(
-                    label="Elementos Específicos",
-                    placeholder="Ex: conflitos, conceitos, exemplos práticos"
-                )
-                público_alvo = gr.Textbox(
-                    label="Público Alvo",
-                    placeholder="Ex: iniciantes, profissionais, estudantes"
-                )
-                tom = gr.Dropdown(
-                    choices=["formal", "informal", "técnico", "conversacional", "educativo"],
-                    label="Tom",
-                    value="formal"
-                )
-                comprimento = gr.Slider(
-                    minimum=100,
-                    maximum=2000,
-                    step=100,
-                    label="Comprimento (palavras)",
-                    value=500
-                )
-                usar_llm = gr.Checkbox(
-                    label="Usar IA para expandir prompt",
-                    value=True
+                with gr.Row():
+                    gerar_btn = gr.Button(
+                        "🚀 Gerar Prompt",
+                        variant="primary",
+                        scale=1,
+                    )
+                
+                with gr.Row():
+                    saida = gr.TextArea(
+                        label="📝 Prompt Gerado",
+                        lines=10,
+                        container=False,
+                    )
+            
+            with gr.TabItem("ℹ️ Sobre"):
+                gr.Markdown(
+                    """
+                    ### Sobre o Gerador de Prompts
+                    
+                    Este é um gerador de prompts inteligente que utiliza IA para criar prompts estruturados
+                    e detalhados para diversos tipos de conteúdo. Ele pode ser usado para:
+                    
+                    - 📚 Criação de histórias
+                    - 📖 Análise de textos
+                    - 💻 Documentação técnica
+                    - 🎓 Conteúdo educacional
+                    
+                    O gerador usa um modelo de linguagem avançado para expandir e melhorar os prompts básicos,
+                    tornando-os mais detalhados e específicos para suas necessidades.
+                    """
                 )
         
-        gerar_btn = gr.Button("Gerar Prompt")
-        saida = gr.Textbox(label="Prompt Gerado", lines=10)
-        
         def atualizar_subcategorias(categoria):
             return gr.Dropdown(choices=list(TEMPLATES[categoria].keys()))
             
@@ -169,8 +211,6 @@ def create_interface():
     
     return app
 
-# Criar e iniciar a aplicação
-app = create_interface()
-
 if __name__ == "__main__":
+    app = create_interface()
     app.launch()