Update app.py

app.py

@@ -1,7 +1,66 @@
+import nltk
+from nltk.chat.util import Chat, reflections
 import gradio as gr
-from huggingface_hub import InferenceClient
 
-client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
+# Reflection to handle variations in user inputs
+reflections = {
+    "aku": "kamu",
+    "saya": "Anda",
+    "kamu": "saya",
+    "Anda": "saya",
+    "adalah": "adalah",
+}
+
+# Define chatbot pairs
+pairs = [
+    [
+        r"(hi|halo|hai|hello|hey)",
+        ["Halo! Saya Nexus-Gerd-Bot. Bagaimana saya bisa membantu Anda hari ini?",]
+    ],
+    [
+        r"siapa kamu\??",
+        ["Saya adalah Nexus-Gerd-Bot, asisten virtual untuk membantu Anda menilai kemungkinan GERD.",]
+    ],
+    [
+        r"bagaimana kabarmu\??",
+        ["Saya baik-baik saja. Terima kasih sudah bertanya!",]
+    ],
+    [
+        r"(.*)gerd(.*)",
+        ["Saya bisa membantu menilai apakah Anda mungkin menderita GERD berdasarkan gejala yang Anda alami dalam 7 hari terakhir. Siapkah Anda untuk memulai?",]
+    ],
+    [
+        r"(.*) (nyeri ulu hati|heartburn) (.*) ?",
+        ["Berapa hari dalam 7 hari terakhir Anda mengalami nyeri ulu hati?",]
+    ],
+    [
+        r"(.*) (regurgitasi) (.*) ?",
+        ["Berapa hari dalam 7 hari terakhir Anda mengalami regurgitasi?",]
+    ],
+    [
+        r"(.*) (mual) (.*) ?",
+        ["Berapa hari dalam 7 hari terakhir Anda mengalami mual?",]
+    ],
+    [
+        r"(.*) (sulit tidur) (.*) ?",
+        ["Berapa hari dalam 7 hari terakhir Anda mengalami kesulitan tidur karena nyeri ulu hati?",]
+    ],
+    [
+        r"(.*) (minum obat) (.*) ?",
+        ["Berapa hari dalam 7 hari terakhir Anda minum obat tambahan untuk nyeri ulu hati?",]
+    ],
+    [
+        r"terima kasih(.*)",
+        ["Sama-sama! Jika Anda membutuhkan bantuan lebih lanjut, jangan ragu untuk bertanya.",]
+    ],
+    [
+        r"selamat tinggal|dadah|sampai jumpa",
+        ["Sampai jumpa! Semoga hari Anda menyenangkan!",]
+    ],
+]
+
+# Create a Chat object
+chatbot = Chat(pairs, reflections)
 
 # Function to convert days to score
 def days_to_score(days):
@@ -20,66 +79,37 @@ def days_to_score(days):
 def diagnose_gerd_responses(responses):
     scores = [days_to_score(d) for d in responses]
     total_score = sum(scores)
+
     if total_score <= 7:
         diagnosis = "Kemungkinan Anda tidak menderita GERD."
     elif 8 <= total_score <= 18:
         diagnosis = "Kemungkinan Anda menderita GERD. Konsultasikan dengan penyedia layanan kesehatan untuk evaluasi lebih lanjut."
     else:
         diagnosis = "Skor di luar jangkauan. Pastikan Anda memasukkan nilai dengan benar."
-    return diagnosis
-
-def respond(message, history, system_message, max_tokens, temperature, top_p):
-    messages = [{"role": "system", "content": system_message}]
-
-    for val in history:
-        if val[0]:
-            messages.append({"role": "user", "content": val[0]})
-        if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
-
-    messages.append({"role": "user", "content": message})
 
-    response = ""
-    user_responses = []
-
-    for message in client.chat_completion(messages, max_tokens=max_tokens, stream=True, temperature=temperature, top_p=top_p):
-        token = message.choices[0].delta.content
-        response += token
-        yield response
-
-    if "nyeri ulu hati" in message:
-        user_responses.append(int(message))
-        response = "Berapa hari dalam 7 hari terakhir Anda mengalami nyeri ulu hati?"
-    elif "regurgitasi" in message:
-        user_responses.append(int(message))
-        response = "Berapa hari dalam 7 hari terakhir Anda mengalami regurgitasi?"
-    elif "mual" in message:
-        user_responses.append(int(message))
-        response = "Berapa hari dalam 7 hari terakhir Anda mengalami mual?"
-    elif "sulit tidur" in message:
-        user_responses.append(int(message))
-        response = "Berapa hari dalam 7 hari terakhir Anda mengalami kesulitan tidur karena nyeri ulu hati?"
-    elif "minum obat" in message:
-        user_responses.append(int(message))
-        response = "Berapa hari dalam 7 hari terakhir Anda minum obat tambahan untuk nyeri ulu hati?"
+    return diagnosis
 
-    if len(user_responses) == 6:
-        diagnosis = diagnose_gerd_responses(user_responses)
-        response = diagnosis
+# Function to handle chatbot responses and collect inputs
+def chatbot_interface(user_input, state):
+    responses = state or []
 
-    yield response
+    if len(responses) < 6:
+        response = chatbot.respond(user_input)
+        responses.append(int(user_input))
+        return response, responses
+    else:
+        diagnosis = diagnose_gerd_responses(responses)
+        return diagnosis, responses
 
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
-        gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
-    ],
+# Define the Gradio interface
+interface = gr.Interface(
+    fn=chatbot_interface,
+    inputs=[gr.Textbox(lines=2, placeholder="Masukkan pesan Anda di sini...")],
+    outputs=[gr.Textbox(label="Nexus-Gerd-Bot"), gr.State()],
     title="Nexus-Gerd-Bot",
     description="Chat dengan Nexus-Gerd-Bot untuk menilai kemungkinan GERD berdasarkan gejala yang Anda alami dalam 7 hari terakhir."
 )
 
+# Launch the app
 if __name__ == "__main__":
-    demo.launch()
+    interface.launch()