Update app.py

app.py

@@ -1,15 +1,25 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 
+"""
+For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
+"""
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 
-# Fungsi untuk merespons pesan
-def respond(message, history, system_message, max_tokens, temperature, top_p):
-    system_message = "Hai, apa ada yang bisa saya bantu?"  # Pesan sistem di awal
 
-    # Menyusun riwayat percakapan
+def respond(
+    message,
+    history: list[tuple[str, str]],
+    system_message,
+    max_tokens,
+    temperature,
+    top_p,
+):
+    # Set system message in Indonesian
+    system_message = "Hai, apa ada yang bisa saya bantu?"
+
     messages = [{"role": "system", "content": system_message}]
-    
+
     for val in history:
         if val[0]:
             messages.append({"role": "user", "content": val[0]})
@@ -19,6 +29,7 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
     messages.append({"role": "user", "content": message})
 
     response = ""
+
     for message in client.chat_completion(
         messages,
         max_tokens=max_tokens,
@@ -27,41 +38,30 @@ def respond(message, history, system_message, max_tokens, temperature, top_p):
         top_p=top_p,
     ):
         token = message.choices[0].delta.content
+
         response += token
         yield response
 
-    # Mengembalikan dua output yang diperlukan oleh Gradio
-    history.append((message, response))  # Menyimpan riwayat percakapan
-    return history, ""  # Mengembalikan history percakapan dan input textbox yang kosong
-
-# Initial message dalam format yang benar
-initial_messages = [{"role": "system", "content": "Hai, apa ada yang bisa saya bantu?"}]
 
-# Membuat antarmuka Gradio
-with gr.Blocks(fill_height=True, css="footer {visibility: hidden}") as demo:
-    gr.HTML("<div style='text-align: center;'><h2>Chatbot</h2></div>")  # Judul
-    
-    chatbot = gr.Chatbot(
-        label="Chatbot",
-        elem_id="chatbot",
-        bubble_full_width=True,
-        scale=1,
-        avatar_images=(None, "https://iili.io/d5tk44a.md.png"),
-        placeholder="Masukkan pesan...",
-        render_markdown=True,
-        sanitize_html=True,
-        show_copy_button=True,
-        value=initial_messages,  # Format yang benar (messages)
-        type="messages"  # Menambahkan 'type="messages"' untuk menghindari peringatan
-    )
-    
-    chat_input = gr.Textbox(interactive=True, placeholder="Masukkan pesan...", show_label=False)
-    
-    # Proses input dan respons
-    chat_msg = chat_input.submit(respond, [chat_input, chatbot], [chatbot, chat_input])
-    bot_msg = chat_msg.then(respond, inputs=[chat_input, chatbot], outputs=[chatbot, chat_input])
+"""
+For information on how to customize the ChatInterface, peruse the gradio docs: https://www.gradio.app/docs/chatinterface
+"""
+demo = gr.ChatInterface(
+    respond,
+    additional_inputs=[
+        gr.Textbox(value="Anda adalah chatbot yang ramah.", label="Pesan Sistem"),
+        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max token baru"),
+        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperatur"),
+        gr.Slider(
+            minimum=0.1,
+            maximum=1.0,
+            value=0.95,
+            step=0.05,
+            label="Top-p (nukleus sampling)",
+        ),
+    ],
+)
 
-    chatbot.like(lambda: "Like/dislike", None, None)  # Fitur like tetap ada
 
-# Menjalankan antarmuka
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()