Update app.py

app.py

@@ -11,11 +11,14 @@ enable_btn = gr.Button.update(interactive=True)
 with open('chatbot_urls.json', 'r') as file:
     chatbots = json.load(file)
 def clear_chat(state):
-    if state is not None:
-        state = {}
-    return state, None,None,gr.Button.update(interactive=False),gr.Button.update(interactive=False),gr.Textbox.update(interactive=False),gr.Button.update(interactive=False)
-
-
+    # Reset state including the chatbot order
+    state = {} if state is not None else state
+    # Shuffle and reinitialize chatbots in the state
+    bot_names = list(chatbots.keys())
+    random.shuffle(bot_names)
+    state['last_bots'] = [bot_names[0], bot_names[1]]
+    # Reset other components
+    return state, None, None, gr.Button.update(interactive=False), gr.Button.update(interactive=False), gr.Textbox.update(interactive=False), gr.Button.update(interactive=False)
 
 
 
@@ -79,15 +82,16 @@ def get_bot_response(url, prompt,state,bot_index):
     return response.json()['output'].split('### Instruction')[0]
 
 def chat_with_bots(user_input, state):
-    bot_names = list(chatbots.keys())
-    random.shuffle(bot_names)
-    bot1_url, bot2_url = chatbots[bot_names[0]], chatbots[bot_names[1]]
+    # Use existing bot order from state if available, otherwise shuffle and initialize
+    if 'last_bots' not in state or not state['last_bots']:
+        bot_names = list(chatbots.keys())
+        random.shuffle(bot_names)
+        state['last_bots'] = [bot_names[0], bot_names[1]]
     
-    # Update the state with the names of the last bots
-    state.update({'last_bots': [bot_names[0], bot_names[1]]})
+    bot1_url, bot2_url = chatbots[state['last_bots'][0]], chatbots[state['last_bots'][1]]
 
-    bot1_response = get_bot_response(bot1_url, user_input,state,0)
-    bot2_response = get_bot_response(bot2_url, user_input,state,1)
+    bot1_response = get_bot_response(bot1_url, user_input, state, 0)
+    bot2_response = get_bot_response(bot2_url, user_input, state, 1)
 
     return bot1_response, bot2_response
 
@@ -202,4 +206,4 @@ with gr.Blocks() as demo:
 
     # Launch the Gradio interface
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()