Update app.py

app.py

@@ -1,31 +1,44 @@
-import gradio as gr
-import os
-import requests
 
-SYSTEM_PROMPT = "O ChatGEO é um assistente virtual especializado em geografia, projetado para ajudar estudantes a explorar e compreender diversos tópicos relacionados ao mundo ao seu redor. Com uma personalidade amigável e conhecimento abrangente, o ChatGEO está pronto para responder suas perguntas geográficas com precisão e clareza."
+SYSTEM_PROMPT = "O ChatGEO é um assistente virtual especializado em geografia, projetado para ajudar estudantes a explorar e compreender diversos tópicos relacionados ao mundo ao seu redor. Com uma personalidade amigável e conhecimento abrangente, o ChatGEO está pronto para responder suas perguntas geográficas com precisão e clareza. Exemplos de Perguntas para ChatGEO: Qual é a capital do Brasil? Descreva o processo de formação de uma cadeia de montanhas. Quais são os principais rios da América do Sul? Explique o conceito de zonas climáticas e suas características. Como a globalização impacta as relações entre os países? Quais são as diferenças entre latitude e longitude? Liste os países que fazem parte da União Europeia. O que são fenômenos naturais e como eles afetam o ambiente?Fale sobre as características geográficas da África Subsaariana. Como os vulcões se formam e quais são seus efeitos na paisagem? Ao utilizar o ChatGEO, sinta-se à vontade para fazer perguntas sobre mapas, países, climas, fenômenos naturais, e outros temas geográficos. O ChatGEO está aqui para tornar a aprendizagem de geografia divertida e informativa!"
 
 TITLE = "ChatGEO"
 EXAMPLE_INPUT = "Olá, ChatGEO!"
+from typing import Iterator
+
+import gradio as gr
+import os
+import requests
+import matplotlib
+matplotlib.use('agg')
 
 zephyr_7b_beta = "https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta/"
+
+
 HF_TOKEN = os.getenv("HF_TOKEN")
 HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
 
 def build_input_prompt(message, chatbot, system_prompt):
-    input_prompt = "\n" + system_prompt + "</s>\n\n"
-    if chatbot:
-        for interaction in chatbot:
-            input_prompt = input_prompt + str(interaction[0]) + "</s>\n\n" + str(interaction[1]) + "\n</s>\n\n"
-    input_prompt = input_prompt + str(message) + "</s>\n"
+    """
+    Constrói a string do prompt de entrada a partir das interações do chatbot e da mensagem atual.
+    """
+    input_prompt = "<|system|>\n" + system_prompt + "</s>\n<|user|>\n"
+    for interaction in chatbot:
+        input_prompt = input_prompt + str(interaction[0]) + "</s>\n<|assistant|>\n" + str(interaction[1]) + "\n</s>\n<|user|>\n"
+
+    input_prompt = input_prompt + str(message) + "</s>\n<|assistant|>"
     return input_prompt
 
+
 def post_request_beta(payload):
+    """
+    Sends a POST request to the predefined Zephyr-7b-Beta URL and returns the JSON response.
+    """
     response = requests.post(zephyr_7b_beta, headers=HEADERS, json=payload)
-    response.raise_for_status()
+    response.raise_for_status()  # Will raise an HTTPError if the HTTP request returned an unsuccessful status code
     return response.json()
 
-def predict_beta(message, chatbot=None, system_prompt=""):
-    chatbot = [] if chatbot is None else chatbot
+
+def predict_beta(message, chatbot=[], system_prompt=""):
     input_prompt = build_input_prompt(message, chatbot, system_prompt)
     data = {
         "inputs": input_prompt
@@ -52,10 +65,11 @@ def predict_beta(message, chatbot=None, system_prompt=""):
 
 def test_preview_chatbot(message, history):
     response = predict_beta(message, history, SYSTEM_PROMPT)
-    text_start = response.rfind("", ) + len("")
+    text_start = response.rfind("<|assistant|>", ) + len("<|assistant|>")
     response = response[text_start:]
     return response
 
+
 welcome_preview_message = f"""
 Bem vindo ao **{TITLE}**! Diga algo como: 
 "{EXAMPLE_INPUT}"
@@ -64,38 +78,9 @@ Um projeto oferecido por <a href="https://mundodageografia.com.br/chatgeo">Mundo
 
 """
 
-# Configurando o layout com cor verde
-custom_css = """
-    body {
-        background-color: #3D9970;
-        font-family: Arial, sans-serif;
-    }
-    .text-box {
-        background-color: #2ECC40;
-        color: #111111;
-    }
-    .input-box {
-        background-color: #2ECC40;
-        color: #111111;
-    }
-    .output-box {
-        background-color: #FFFFFF;
-        color: #111111;
-    }
-    .output-box a {
-        color: #0074cc;
-    }
-"""
+chatbot_preview = gr.Chatbot(layout="panel", value=[(None, welcome_preview_message)])
+textbox_preview = gr.Textbox(scale=7, container=False, value=EXAMPLE_INPUT)
+
+demo = gr.ChatInterface(test_preview_chatbot, chatbot=chatbot_preview, textbox=textbox_preview)
 
-# Criando a interface Gradio
-demo = gr.Interface(
-    fn=test_preview_chatbot,
-    inputs=gr.Textbox(scale=7, placeholder=EXAMPLE_INPUT, label="Digite sua mensagem:"),
-    outputs=gr.Textbox(scale=7, label="Resposta do ChatGEO:"),
-    live=True,
-    examples=[[EXAMPLE_INPUT]],
-    title=f"<span style='color: #FFFFFF;'>{TITLE}</span>",
-    css=custom_css
-)
-
-demo.launch()
+demo.launch()