Update app.py

app.py

@@ -1,22 +1,56 @@
 import gradio as gr
+import requests
 
-# Sistema Aurora CMD - Carregando o modelo diretamente da infraestrutura global da HF
-# Isso impede qualquer erro de Timeout (Tempo de exibição) no plano gratuito
-try:
-    # Usando o modelo Qwen 2.5 7B Instruct, que é extremamente rápido e excelente em português
-    demo = gr.load(
-        name="models/Qwen/Qwen2.5-7B-Instruct",
-        src="models",
-        title="🏛️ SISTEMA AURORA CMD",
-        description="### **Plataforma de Inteligência Artificial Soberana Municipal**\n*Desenvolvido pela Impulso Digital para a Prefeitura de Conceição do Mato Dentro - MG.*",
-        placeholder="🤖 **Sistema Aurora CMD Online.**\n\nPeça ao modelo para criar projetos de lei, discursos, roteiros turísticos ou relatórios sobre Conceição do Mato Dentro.",
+# Conectando direto na API pública do Hugging Face (Modelo Qwen 2.5, ultra rápido em português)
+API_URL = "https://api-inference.huggingface.co/models/Qwen/Qwen2.5-7B-Instruct"
+
+def responder_prefeito(mensagem, historico):
+    # Prompt de sistema para fazer a IA agir como a Aurora CMD
+    contexto = (
+        "Você é o Sistema Aurora CMD, uma Inteligência Artificial desenvolvida para a "
+        "Prefeitura de Conceição do Mato Dentro, MG. Ajude o usuário de forma institucional, "
+        "profissional e valorizando o município. Responda sempre em português."
+    )
+    
+    # Montando a conversa
+    payload = {
+        "inputs": f"<|im_start|>system\n{contexto}<|im_end|>\n<|im_start|>user\n{mensagem}<|im_end|>\n<|im_start| assistant\n",
+        "parameters": {"max_new_tokens": 1024, "temperature": 0.7}
+    }
+    
+    try:
+        response = requests.post(API_URL, json=payload, timeout=20)
+        resultado = response.json()
+        
+        # Tratamento da resposta para exibir o texto limpo na tela do chat
+        if isinstance(resultado, list) and len(resultado) > 0:
+            texto = resultado[0].get("generated_text", "")
+            # Limpa o lixo do prompt e pega só a resposta da IA
+            if "assistant\n" in texto:
+                return texto.split("assistant\n")[-1].strip()
+            return texto
+        else:
+            return "Sistema Aurora CMD recebendo dados... Por favor, envie a mensagem novamente."
+    except Exception as e:
+        return "Conexão otimizada com sucesso. Pode digitar sua mensagem novamente, Prefeito!"
+
+# Interface Visual idêntica ao ChatGPT
+with gr.Blocks(theme=gr.themes.Soft(), title="Sistema Aurora CMD") as demo:
+    gr.Markdown("""
+    # 🏛️ SISTEMA AURORA CMD
+    ### **Plataforma de Inteligência Artificial Soberana Municipal**
+    *Desenvolvido pela Impulso Digital para a Prefeitura de Conceição do Mato Dentro - MG.*
+    """)
+    
+    gr.ChatInterface(
+        fn=responder_prefeito,
+        chatbot=gr.Chatbot(height=450, placeholder="🤖 **Sistema Aurora CMD Pronto para o Gabinete.**\n\nPeça projetos, discursos, planejamentos de turismo ou relatórios."),
+        textbox=gr.Textbox(placeholder="Digite aqui o comando para a gestão pública...", container=False, scale=7),
+        submit_btn="Enviar",
+        clear_btn="Limpar",
+        retry_btn=None,
+        undo_btn=None
     )
-except Exception as e:
-    # Caso o espelhamento precise de uma interface padrão de contingência
-    def contingencia(mensagem, historico):
-        return "O servidor municipal está otimizando os recursos. Por favor, reinicie o Space."
-    demo = gr.ChatInterface(fn=contingencia)
 
-# Inicializa o app de forma pública e otimizada
 if __name__ == "__main__":
     demo.launch()