Update app.py

app.py

@@ -1,4 +1,3 @@
-import gradio as gr
 import requests
 import os
 import json
@@ -130,17 +129,6 @@ class HuggingFaceAPIClient:
 # Inicializar cliente da API
 api_client = HuggingFaceAPIClient(HF_TOKEN)
 
-def escape_html(text: str) -> str:
-    """Escapa caracteres especiais HTML para exibição segura."""
-    html_escape_table = {
-        "&": "&",
-        "<": "&lt;",
-        ">": "&gt;",
-        '"': "&quot;",
-        "'": "&#x27;"
-    }
-    return "".join(html_escape_table.get(c, c) for c in text)
-
 def formatar_resposta_com_codigo(resposta: str) -> str:
     """Formata a resposta destacando códigos em blocos separados"""
     if not resposta:
@@ -149,7 +137,7 @@ def formatar_resposta_com_codigo(resposta: str) -> str:
     # Detecta blocos de código com ```
     resposta_formatada = re.sub(
         r'```(\w+)?\n(.*?)\n```',
-        lambda m: f'<div style="background-color: #f8f9fa; color: #1a1a1a; border: 1px solid #e9ecef; border-radius: 8px; padding: 15px; margin: 10px 0; font-family: Monaco, Consolas, monospace; overflow-x: auto;"><strong style="color: #1a1a1a;">💻 Código:</strong><br><pre style="color: #1a1a1a; margin: 5px 0; white-space: pre-wrap; word-wrap: break-word;"><code>{escape_html(m.group(2))}</code></pre></div>',
+        r'<div style="background-color: #f8f9fa; color: #1a1a1a; border: 1px solid #e9ecef; border-radius: 8px; padding: 15px; margin: 10px 0; font-family: Monaco, Consolas, monospace; overflow-x: auto;"><strong style="color: #1a1a1a;">💻 Código:</strong><br><pre style="color: #1a1a1a; margin: 5px 0; white-space: pre-wrap; word-wrap: break-word;"><code>\2</code></pre></div>',
         resposta,
         flags=re.DOTALL
     )
@@ -157,7 +145,7 @@ def formatar_resposta_com_codigo(resposta: str) -> str:
     # Detecta código inline com `
     resposta_formatada = re.sub(
         r'`([^`]+)`',
-        lambda m: f'<code style="background-color: #f1f3f4; color: #1a1a1a; padding: 2px 4px; border-radius: 4px; font-family: Monaco, Consolas, monospace;">{escape_html(m.group(1))}</code>',
+        r'<code style="background-color: #f1f3f4; color: #1a1a1a; padding: 2px 4px; border-radius: 4px; font-family: Monaco, Consolas, monospace;">\1</code>',
         resposta_formatada
     )
     
@@ -259,8 +247,8 @@ css_customizado = """
 }
 
 .resposta-container {
-    background-color: #ffffff !important;
-    color: #1a1a1a !important;
+    background-color: #ffffff !important; /* Always light background */
+    color: #1a1a1a !important; /* Always dark text */
     border: 1px solid #e0e0e0 !important;
     border-radius: 20px !important;
     padding: 20px !important;
@@ -269,8 +257,8 @@ css_customizado = """
 }
 
 .resposta-container pre code {
-    color: #1a1a1a !important;
-    background-color: #f8f9fa !important;
+    color: #1a1a1a !important; /* Ensure code text is dark */
+    background-color: #f8f9fa !important; /* Light background for code */
 }
 
 .pergunta-container {
@@ -287,24 +275,6 @@ css_customizado = """
     margin-bottom: 20px !important;
     text-align: center !important;
 }
-
-.modelo-select {
-    background-color: #f0f8ff !important;
-    color: #1a1a1a !important;
-    border: 1px solid #e0e0e0 !important;
-    border-radius: 8px !important;
-    padding: 10px !important;
-    font-size: 14px !important;
-    width: 100% !important;
-    margin-top: 10px !important;
-    font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Oxygen, Ubuntu, Cantarell, 'Open Sans', 'Helvetica Neue', sans-serif !important;
-}
-
-.modelo-select:focus {
-    outline: none !important;
-    border-color: #667eea !important;
-    box-shadow: 0 0 5px rgba(102, 126, 234, 0.5) !important;
-}
 """
 
 # Interface Gradio
@@ -334,22 +304,18 @@ with gr.Blocks(
                 elem_classes="pergunta-container"
             )
             
-            # Renderiza um <select> com HTML
-            modelo_select = gr.HTML(
-                value=f"""
-                <select id="modelo-select" class="modelo-select">
-                    {''.join(f'<option value="{key}" {"selected" if key == "phi3-mini" else ""}>{key}</option>' for key in MODELS.keys())}
-                </select>
-                """,
-                label=""
+            modelo_select = gr.Dropdown(
+                choices=list(MODELS.keys()),
+                value="phi3-mini",
+                label="🧠 Modelo de IA",
+                info="Escolha o modelo para responder sua pergunta"
             )
             
             with gr.Row():
-                botao_perguntar.click(
-                    fn=responder_como_aldo, 
-                    inputs=[entrada, modelo_select],
-                    outputs=saida,
-                    show_progress=True
+                botao_perguntar = gr.Button(
+                    "🤔 Perguntar ao Dr. Aldo", 
+                    variant="primary", 
+                    size="lg"
                 )
                 botao_testar = gr.Button(
                     "🔍 Testar Modelos",