Update app.py

app.py

@@ -7,21 +7,23 @@ client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
 # Initialize conversation history and difficulty
 conversation_history = []
-current_difficulty = "medium"
+current_difficulty = "sedang"
 
 def get_system_message(difficulty):
     base_message = """
-You are an AI assistant designed to simulate customer interactions for retail sales training.
-Your goal is to role-play as a customer with a complaint. The human will play the role of a retail sales representative.
-Evaluate the representative's responses and decide whether to continue the conversation or end it.
-If the representative handles the complaint well, respond positively and indicate the conversation is successful.
-If the representative fails to address the concerns adequately, respond negatively and indicate you wish to disengage.
-Start the conversation by presenting a retail-related complaint.
+Anda adalah asisten AI yang dirancang untuk mensimulasikan pelanggan dalam skenario pelatihan penjualan asuransi.
+Peran Anda HANYA untuk berperan sebagai pelanggan dengan keluhan. JANGAN PERNAH merespons seolah-olah Anda adalah perwakilan penjualan.
+Manusia yang berinteraksi dengan Anda akan berperan sebagai perwakilan penjualan asuransi.
+Evaluasi respons perwakilan dan putuskan apakah akan melanjutkan percakapan atau mengakhirinya.
+Jika perwakilan menangani keluhan Anda dengan baik, tanggapi secara positif sebagai pelanggan yang puas.
+Jika perwakilan gagal mengatasi kekhawatiran Anda dengan memadai, tanggapi secara negatif dan tunjukkan bahwa Anda ingin mengakhiri percakapan.
+Selalu tetap berperan sebagai pelanggan sepanjang percakapan.
+Gunakan Bahasa Indonesia dalam semua respons Anda.
 """
     difficulty_adjustments = {
-        "easy": "Be an easy-going customer, open to suggestions and quick to accept reasonable solutions.",
-        "medium": "Be a moderately challenging customer, requiring some convincing but eventually accepting good solutions.",
-        "hard": "Be a difficult customer, skeptical of solutions and requiring exceptional service to be satisfied."
+        "mudah": "Jadilah pelanggan yang mudah diatur, terbuka terhadap saran, dan cepat menerima solusi yang masuk akal.",
+        "sedang": "Jadilah pelanggan yang cukup menantang, membutuhkan sedikit bujukan tetapi akhirnya menerima solusi yang baik.",
+        "sulit": "Jadilah pelanggan yang sulit, skeptis terhadap solusi dan membutuhkan layanan luar biasa untuk merasa puas."
     }
     return base_message + difficulty_adjustments[difficulty]
 
@@ -34,90 +36,102 @@ def chat_with_gpt(user_message, history):
             *conversation_history
         ]
         response = client.chat.completions.create(
-            model="gpt-3.5-turbo",
+            model="gpt-4o-mini",
             messages=messages,
             max_tokens=150
         )
         bot_response = response.choices[0].message.content.strip()
+        
+        if "sebagai perwakilan penjualan" in bot_response.lower() or "saya bisa membantu anda dengan itu" in bot_response.lower():
+            bot_response = "Maaf, tapi saya adalah pelanggan dalam skenario ini. Bisakah Anda menangani keluhan saya?"
+        
         conversation_history.append({"role": "assistant", "content": bot_response})
-        conversation_ended = "conversation ended" in bot_response.lower() or "disengage" in bot_response.lower()
+        conversation_ended = "percakapan berakhir" in bot_response.lower() or "mengakhiri" in bot_response.lower()
         history.append((user_message, bot_response))
         if conversation_ended:
             summary_feedback = generate_summary_feedback()
             return history, history, "", summary_feedback
         return history, history, "", None
     except Exception as e:
-        error_msg = f"An error occurred in chat_with_gpt: {str(e)}"
+        error_msg = f"Terjadi kesalahan dalam chat_with_gpt: {str(e)}"
         print(error_msg)
         return history + [("Error", error_msg)], history + [("Error", error_msg)], "", None
 
 def generate_summary_feedback():
     try:
-        summary_prompt = "Summarize the conversation, evaluate the sales representative's performance, and provide feedback for improvement. Be concise but comprehensive."
+        summary_prompt = "Rangkum percakapan, evaluasi kinerja perwakilan penjualan, dan berikan umpan balik untuk perbaikan. Buat ringkas namun komprehensif."
         messages = [
-            {"role": "system", "content": "You are an AI assistant providing feedback on a retail sales interaction."},
+            {"role": "system", "content": "Anda adalah asisten AI yang memberikan umpan balik tentang interaksi penjualan ritel."},
             *conversation_history,
             {"role": "user", "content": summary_prompt}
         ]
         response = client.chat.completions.create(
-            model="gpt-3.5-turbo",
+            model="gpt-4o-mini",
             messages=messages,
             max_tokens=250
         )
-        return "Status: Completed\n\n" + response.choices[0].message.content.strip()
+        return "Status: Selesai\n\n" + response.choices[0].message.content.strip()
     except Exception as e:
-        return f"Error generating summary: {str(e)}"
+        return f"Kesalahan dalam menghasilkan ringkasan: {str(e)}"
 
 def clear_conversation():
     global conversation_history
     conversation_history = []
     return None, None, "", None
 
-def start_new_conversation(difficulty):
-    global current_difficulty
+def start_new_conversation(difficulty, scenario):
+    global current_difficulty, conversation_history
     current_difficulty = difficulty
     try:
         clear_conversation()
-        initial_response = client.chat.completions.create(
+        system_message = get_system_message(difficulty)
+        initial_prompt = f"Anda adalah pelanggan dengan keluhan berikut: {scenario}. Mulailah percakapan dengan menyampaikan keluhan ini kepada perwakilan penjualan."
+        
+        conversation_history = [
+            {"role": "system", "content": system_message},
+            {"role": "user", "content": initial_prompt}
+        ]
+        
+        response = client.chat.completions.create(
             model="gpt-3.5-turbo",
-            messages=[
-                {"role": "system", "content": get_system_message(difficulty)},
-                {"role": "user", "content": f"Start a new {difficulty} retail complaint scenario."}
-            ],
+            messages=conversation_history,
             max_tokens=150
-        ).choices[0].message.content.strip()
+        )
+        initial_response = response.choices[0].message.content.strip()
         conversation_history.append({"role": "assistant", "content": initial_response})
+        
         return [(None, initial_response)], [(None, initial_response)], "", None
     except Exception as e:
-        error_msg = f"An error occurred in start_new_conversation: {str(e)}"
+        error_msg = f"Terjadi kesalahan dalam memulai percakapan baru: {str(e)}"
         print(error_msg)
         return [("Error", error_msg)], [("Error", error_msg)], "", None
 
 def end_conversation(history):
     global conversation_history
-    conversation_history.append({"role": "user", "content": "The sales representative has ended the conversation."})
+    conversation_history.append({"role": "user", "content": "Perwakilan penjualan telah mengakhiri percakapan."})
     summary_feedback = generate_summary_feedback()
-    history.append((None, "The sales representative has ended the conversation."))
+    history.append((None, "Perwakilan penjualan telah mengakhiri percakapan."))
     return history, history, "", summary_feedback
 
 # Set up Gradio interface
 with gr.Blocks() as demo:
-    gr.Markdown("# Retail Sales Training Simulator")
-    gr.Markdown("Select the difficulty level and start a new scenario to begin.")
+    gr.Markdown("# Simulator Pelatihan Penjualan Ritel")
+    gr.Markdown("Pilih tingkat kesulitan, masukkan skenario, dan mulai simulasi untuk memulai. Anda akan berperan sebagai perwakilan penjualan.")
 
-    difficulty_radio = gr.Radio(["easy", "medium", "hard"], label="Customer Difficulty", value="medium")
-    new_scenario = gr.Button("Start New Scenario")
+    difficulty_radio = gr.Radio(["mudah", "sedang", "sulit"], label="Tingkat Kesulitan Pelanggan", value="sedang")
+    scenario_input = gr.Textbox(label="Masukkan skenario keluhan pelanggan", lines=3)
+    new_scenario = gr.Button("Mulai Simulasi")
 
     chatbot = gr.Chatbot()
-    msg = gr.Textbox(label="Your response")
+    msg = gr.Textbox(label="Respons Anda sebagai perwakilan penjualan")
     with gr.Row():
-        submit_btn = gr.Button("Submit")
-        end_btn = gr.Button("End Conversation")
+        submit_btn = gr.Button("Kirim")
+        end_btn = gr.Button("Akhiri Percakapan")
 
-    summary = gr.Textbox(label="Conversation Summary and Feedback", lines=10, interactive=False)
-    clear = gr.Button("Clear Conversation")
+    summary = gr.Textbox(label="Ringkasan Percakapan dan Umpan Balik", lines=100, interactive=False)
+    clear = gr.Button("Bersihkan Percakapan")
 
-    new_scenario.click(start_new_conversation, inputs=[difficulty_radio], outputs=[chatbot, chatbot, msg, summary])
+    new_scenario.click(start_new_conversation, inputs=[difficulty_radio, scenario_input], outputs=[chatbot, chatbot, msg, summary])
     submit_btn.click(chat_with_gpt, [msg, chatbot], [chatbot, chatbot, msg, summary])
     msg.submit(chat_with_gpt, [msg, chatbot], [chatbot, chatbot, msg, summary])
     end_btn.click(end_conversation, inputs=[chatbot], outputs=[chatbot, chatbot, msg, summary])