Update app.py

app.py

@@ -114,6 +114,11 @@ def emotion_analysis():
     bot_emotion=sentiment.emotion_analysis(bot_conversation)
     return "Emotions for customer_conversation:\n"+customer_emotion+"\nEmotions for sales_agent_conversation:\n"+bot_emotion
 
+def clear_stages():
+    global stage
+    stage = ""
+    return ""
+    
 with gr.Blocks(theme="Taithrah/Minimal") as demo:
     gr.HTML("""<center><h1>Sales Persona Chatbot</h1></center>""")
     
@@ -163,6 +168,7 @@ with gr.Blocks(theme="Taithrah/Minimal") as demo:
     emotion_btn.click(emotion_analysis,[],emotion_view)
     clear.click(lambda: None, None, chatbot, queue=False)
     clear.click(lambda: None, None, show_stages, queue=False)
+    clear.click(clear_stages,[],show_stages)
     
 demo.queue()
 demo.launch()