profdanielvieira95 commited on
Commit
ea89da4
·
verified ·
1 Parent(s): 99878bb

codigo chatbot

Files changed (1) hide show
  1. app.py +44 -0
app.py ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os # Biblioteca para interagir com o sistema operacional
2
+ import gradio as gr # Importa Gradio para criar a interface web
3
+ from groq import Groq # Importa o cliente da API Groq
4
+
5
+ # Defina sua chave da API do GROQ
6
+ GROQ_API_KEY = os.getenv('GROQ_API_KEY', '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') # Substitua pela sua chave
7
+
8
+ # Inicializa o cliente da API Groq
9
+ client = Groq(api_key=GROQ_API_KEY)
10
+
11
+ # Função para processar a entrada do usuário e gerar uma resposta
12
+ def assistente_agricultura(user_prompt):
13
+ if user_prompt.strip() == "15":
14
+ return "Encerrando assistente Agro! Até mais! 🚜"
15
+
16
+ # Faz a solicitação ao modelo
17
+ completion = client.chat.completions.create(
18
+ model="llama3-8b-8192",
19
+ messages=[
20
+ {"role": "system", "content": "Você é um assistente especializado em agricultura."},
21
+ {"role": "user", "content": user_prompt}
22
+ ],
23
+ temperature=0,
24
+ max_tokens=1024,
25
+ top_p=1,
26
+ stream=False
27
+ )
28
+
29
+ # Retorna a resposta do assistente
30
+ return completion.choices[0].message.content
31
+
32
+ # Configuração da interface Gradio
33
+ iface = gr.Interface(
34
+ fn=assistente_agricultura,
35
+ inputs=gr.Textbox(lines=2, placeholder="Digite sua pergunta sobre agricultura..."),
36
+ outputs="text",
37
+ title="🌾 Assistente Agro IA",
38
+ description="Digite sua pergunta sobre agricultura e receba respostas de IA especializadas! 🚜",
39
+ live=True
40
+ )
41
+
42
+ # Executa a interface web
43
+ if __name__ == "__main__":
44
+ iface.launch()