Update app.py

app.py

@@ -28,21 +28,24 @@ def hive_response(query, bots_config_json):
         # Collect responses from all bots
         responses = [future.result() for future in futures]
 
-    return [f"Bot {idx} response: {response}" for idx, response in enumerate(responses, 1)]
+    return [f"{bot['role']} Bot response: {response}" for bot, response in zip(bots_config, responses)]
 
 # Define the Gradio interface
+default_bots_config = json.dumps([
+    {"role": "HR", "tune-adafd1fc-f66f-4242-aed1-c0ce3722718a1711930019": "tune-hr-api-key", "base_url": "https://chat.tune.app/api/chat/completions", "model_name": "goliath-120b-16k-gptq"},
+    {"role": "IT", "tune-adafd1fc-f66f-4242-aed1-c0ce3722718a1711930019": "tune-it-api-key", "base_url": "https://chat.tune.app/api/chat/completions", "model_name": "goliath-120b-16k-gptq"},
+    {"role": "Sales", "tune-adafd1fc-f66f-4242-aed1-c0ce3722718a1711930019": "tune-sales-api-key", "base_url": "https://chat.tune.app/api/chat/completions", "model_name": "goliath-120b-16k-gptq"},
+    {"role": "Customer Support", "tune-adafd1fc-f66f-4242-aed1-c0ce3722718a1711930019": "tune-cs-api-key", "base_url": "https://chat.tune.app/api/chat/completions", "model_name": "goliath-120b-16k-gptq"}
+], indent=2)
+
 iface = gr.Interface(
     fn=hive_response,
     inputs=[
         gr.Textbox(label="Query"),
-        gr.Textbox(label="Bots Configuration (JSON)", default=json.dumps([
-            {"api_key": "tune-adafd1fc-f66f-4242-aed1-c0ce3722718a1711930019", "base_url": "https://chat.tune.app/api/", "model_name": "goliath-120b-16k-gptq"},
-            {"api_key": "tune-bbb2d24b-f913-48d3-b3fe-dcf2b8ed10371711926291", "base_url": "https://chat.tune.app/api/", "model_name": "goliath-120b-16k-gptq"},
-            {"api_key": "tune-a67090f0-e25e-4769-a67e-86d71ba22e0f1696395693", "base_url": "https://chat.tune.app/api/", "model_name": "goliath-120b-16k-gptq"}
-        ], indent=2))
+        gr.Textbox(label="Bots Configuration (JSON)", default=default_bots_config)
     ],
     outputs=[gr.Text(label="Responses")],
-    description="Enter your query and bots configuration in JSON format to get responses from a hive of chatbots."
+    description="Enter your query and configure the bots in JSON format to simulate interactions with different company departments."
 )
 
 iface.launch()