Create interface.py

interface.py

@@ -0,0 +1,133 @@
+import gradio as gr
+from ai_logic import (
+    responder_como_aldo, 
+    build_and_save_vector_store, 
+    testar_todos_modelos,
+    MODELS, 
+    DEFAULT_MODEL
+)
+
+# CSS personalizado
+css_customizado = """
+.gradio-container { max-width: 1400px !important; margin: 0 auto; width: 99%; }
+.gr-textbox textarea { font-size: 14px !important; line-height: 1.5 !important; }
+.resposta-container { background-color: #ffffff !important; color: #1a1a1a !important; border: 1px solid #e0e0e0 !important; border-radius: 20px !important; padding: 20px !important; margin: 20px 0 !important; box-shadow: 0 2px 4px rgba(0, 0, 0, 0.05) !important; }
+.resposta-container pre code { color: #1a1a1a !important; background-color: #f8f9fa !important; }
+.pergunta-container { background-color: #f0f8ff !important; border-radius: 8px !important; padding: 15px !important; }
+.titulo-principal { background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important; color: white !important; padding: 20px !important; border-radius: 10px !important; margin-bottom: 20px !important; text-align: center !important; }
+.modelo-dropdown { margin-bottom: 15px !important; }
+"""
+
+def criar_interface():
+    """Cria e retorna a interface Gradio"""
+    
+    with gr.Blocks(title="Dr. Aldo Henrique - API Externa", theme=gr.themes.Soft(), css=css_customizado) as interface:
+        # Cabeçalho
+        gr.HTML("""
+        <div class="titulo-principal">
+            <h1>🤖 Dr. Aldo Henrique - Foco em TI com diferentes modelos de IA </h1>
+            <p style="font-size: 14px; opacity: 0.9;">Conhecimento enriquecido com o conteúdo do <a href="https://aldohenrique.com.br/" style="color: white; text-decoration: underline;">Blog do Prof. Dr. Aldo Henrique</a></p>
+        </div>
+        """)
+        
+        # Interface principal
+        with gr.Row():
+            # Coluna de entrada
+            with gr.Column(scale=2):
+                gr.Markdown("### 📝 Faça sua pergunta:")
+                entrada = gr.Textbox(
+                    label="", 
+                    placeholder="Digite sua pergunta aqui.", 
+                    lines=6, 
+                    elem_classes="pergunta-container"
+                )
+                modelo_select = gr.Dropdown(
+                    choices=list(MODELS.keys()), 
+                    value=DEFAULT_MODEL, 
+                    label="🧠 Selecione o Modelo de IA", 
+                    info="Escolha o modelo para responder", 
+                    elem_classes="modelo-dropdown"
+                )
+                
+                with gr.Row():
+                    botao_perguntar = gr.Button("🤔 Perguntar ao Dr. Aldo", variant="primary", size="lg")
+                    #botao_testar = gr.Button("🔍 Testar Modelos", variant="secondary")
+
+            # Coluna de saída
+            with gr.Column(scale=3):
+                gr.Markdown("### 💬 Resposta do Dr. Aldo Henrique:")
+                saida = gr.HTML(
+                    label="", 
+                    value="<div style='padding: 20px; text-align: center; color: #1a1a1a;'>Aguardando sua pergunta...</div>", 
+                    elem_classes="resposta-container"
+                )
+
+        # --- SEÇÃO PARA CONTROLE DO RAG ---
+        with gr.Accordion("⚙️ Controle do Conhecimento (RAG)", open=False):
+            status_rag = gr.Textbox(label="Status do Retreino", interactive=False)
+            botao_retreinar = gr.Button("🔄 Atualizar Conhecimento do Blog", variant="stop")
+            # Componentes para download
+            download_faiss_file = gr.File(
+                label="Download do Índice FAISS", 
+                interactive=False, 
+                file_count="single", 
+                file_types=[".pkl"]
+            )
+            download_urls_file = gr.File(
+                label="Download das URLs Processadas", 
+                interactive=False, 
+                file_count="single", 
+                file_types=[".pkl"]
+            )
+
+        # Exemplos
+        with gr.Accordion("📚 Exemplos de Perguntas", open=False):
+            gr.Examples(
+                examples=[
+                    ["Como implementar uma lista ligada em C com todas as operações básicas?", DEFAULT_MODEL],
+                    ["Qual a sua opinião sobre o uso de ponteiros em C++ moderno, baseada no seu blog?", "Mistral 7B"],
+                    ["Resuma o que você escreveu sobre machine learning no seu blog.", "Zephyr 7B"],
+                ],
+                inputs=[entrada, modelo_select]
+            )
+        
+        # Status da API
+        with gr.Accordion("🔧 Status da API", open=False):
+            status_api = gr.Textbox(label="Status dos Modelos", interactive=False, lines=8)
+        
+        # Informações
+        with gr.Accordion("ℹ️ Informações", open=False):
+            gr.Markdown("""
+            ### Sobre o Dr. Aldo Henrique:
+            - **Especialidade**: Linguagens C, Java, Desenvolvimento Web, Inteligência Artificial
+            - **Conhecimento Adicional**: Conteúdo do blog aldohenrique.com.br
+            ### Dicas para melhores respostas:
+            - Faça perguntas específicas sobre o conteúdo do blog para ver o RAG em ação!
+            - Peça resumos ou opiniões sobre temas que o professor aborda.
+            """)
+            
+        # Configuração dos eventos
+        botao_perguntar.click(
+            fn=responder_como_aldo, 
+            inputs=[entrada, modelo_select], 
+            outputs=saida, 
+            show_progress=True
+        )
+        
+        #botao_testar.click(
+        #    fn=testar_todos_modelos, 
+        #    outputs=status_api, 
+        #    show_progress=True
+        #)
+        
+        botao_retreinar.click(
+            fn=build_and_save_vector_store,
+            outputs=[status_rag, download_faiss_file, download_urls_file],
+            show_progress=True
+        )
+    
+    return interface
+
+def configurar_interface():
+    """Configura e retorna a interface pronta para lançamento"""
+    return criar_interface()