Update functions.py

functions.py

@@ -1,16 +1,14 @@
 import plotly.express as px
-import gradio as gr
 from client import client
 
 # Fungsi untuk menghasilkan plot acak
 def random_plot():
     df = px.data.iris()
-    fig = px.scatter(df, x="sepal_width", y="sepal_length", color="species",
-                    size='petal_length', hover_data=['petal_width'])
+    fig = px.scatter(df, x="sepal_width", y="sepal_length", color="species", size='petal_length', hover_data=['petal_width'])
     return fig
 
 # Fungsi untuk mencetak data suka/tidak suka
-def print_like_dislike(x: gr.LikeData):
+def print_like_dislike(x):
     print(x.index, x.value, x.liked)
 
 # Fungsi untuk menambahkan pesan ke riwayat
@@ -23,8 +21,8 @@ def add_message(history, message):
     return history, gr.Textbox(value="", interactive=True)
 
 # Fungsi untuk menghubungkan dengan model Llama
-def chat_mem(message, chat_history):
-    chat_history_role = [{"role": "system", "content": "Anda adalah Chat Bot milik Riswan Nopiyar, yang dirancang untuk membantu menjawab pertanyaan dan memberikan informasi dengan sopan dan akurat dalam bahasa Indonesia. Seluruh percakapan akan dilakukan dalam bahasa Indonesia. apapun kondisinya kamu harus tetap menggunakan bahasa indonesia meskipun jika kamu kurang menguasai"}]
+def chat_mem(message, chat_history, max_tokens=512):
+    chat_history_role = [{"role": "system", "content": "Anda adalah Chat Bot milik Riswan Nopiyar, yang dirancang untuk membantu menjawab pertanyaan dan memberikan informasi dengan sopan dan akurat dalam bahasa Indonesia. Seluruh percakapan akan dilakukan dalam bahasa Indonesia."}]
     
     if chat_history:
         for user_msg, assistant_msg in chat_history:
@@ -38,7 +36,7 @@ def chat_mem(message, chat_history):
     else:
         chat_completion = client.chat_completion(
             messages=chat_history_role,
-            max_tokens=2048,
+            max_tokens=max_tokens,
         )
         assistant_response = chat_completion.choices[0].message.content
     
@@ -47,4 +45,4 @@ def chat_mem(message, chat_history):
     modified = [msg["content"] for msg in chat_history_role]
     chat_history = [(modified[i * 2 + 1], modified[i * 2 + 2]) for i in range(len(modified) // 2)]
 
-    return "", chat_history
+    return "", chat_history