Create app.py

app.py

@@ -0,0 +1,70 @@
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
+
+import pandas as pd
+df_status_pedidos = pd.DataFrame({
+    "numero_pedido": [12345, 67890, 54321],
+    "status": ["Em processamento", "Enviado", "Entregue"]
+})
+
+def verificar_status_pedido(numero_pedido):
+    try:
+        numero_pedido = int(numero_pedido)
+
+        print(f"Procurando o status para o pedido número: {numero_pedido}")
+        if numero_pedido in df_status_pedidos["numero_pedido"].values:
+            status = df_status_pedidos[df_status_pedidos["numero_pedido"] == numero_pedido]["status"].iloc[0]
+            return f"O status do pedido {numero_pedido} é: {status}"
+        else:
+            return "Número do pedido não encontrado. Verifique e tente novamente."
+
+    except ValueError:
+        return "Número do pedido inválido. Insira um valor numérico."
+
+palavras_chave_status = ["order", "order status", "status of my order", "check my order", "track my order", "order update"]
+
+esperando_numero_pedido = False
+
+def chatbot_response(usuario):
+    global esperando_numero_pedido
+
+    if usuario.lower() in ["quit", "exit", "bye", "stop"]:
+        return "Goodbye!"
+
+    if esperando_numero_pedido:
+        resposta = verificar_status_pedido(usuario)
+        esperando_numero_pedido = False
+        return resposta
+
+    if any(keyword in usuario.lower() for keyword in palavras_chave_status):
+        esperando_numero_pedido = True
+        return "Por favor, forneça o número do pedido."
+
+    historico_chat = None
+    novo_usuario_input_ids = tokenizer.encode(usuario + tokenizer.eos_token, return_tensors='pt')
+    if historico_chat is not None:
+        bot_input_ids = torch.cat([historico_chat, novo_usuario_input_ids], dim=-1)
+    else:
+        bot_input_ids = novo_usuario_input_ids
+
+    historico_chat = model.generate(
+        bot_input_ids,
+        max_length=1000,
+        pad_token_id=tokenizer.eos_token_id
+    )
+    resposta = tokenizer.decode(historico_chat[:, bot_input_ids.shape[-1]:][0], skip_special_tokens=True)
+
+    return resposta
+
+app = gr.Interface(
+    fn=chatbot_response,
+    inputs="text",
+    outputs="text",
+    title="Chatbot com Verificação de Status",
+    description="Digite sua mensagem para conversar com o bot. Pergunte sobre o status do pedido ou faça outra pergunta."
+)
+
+if __name__ == "__main__":
+    app.launch()