Create app.py

app.py

@@ -0,0 +1,32 @@
+import streamlit as st
+from huggingface_hub import InferenceClient
+
+client = InferenceClient("mistralai/Mixtral-8x7B-Instruct-v0.1")
+
+def format_prompt(agent_id, message):
+    return f"<s><Agent {agent_id}>[INST] {message} [/INST]"
+
+def generate_response(agent_id, message):
+    formatted_prompt = format_prompt(agent_id, message)
+    response = client.text_generation(formatted_prompt, stream=False)
+    return response['generated_text']
+
+def generate_group_therapy_responses(message):
+    responses = []
+    for agent_id in range(1, 5):  # Four agents
+        agent_response = generate_response(agent_id, message)
+        responses.append(f"Agent {agent_id}: {agent_response}")
+    return responses
+
+st.title("Group Therapy Simulation")
+st.write("Ask a question and receive perspectives from four different agents.")
+
+user_question = st.text_input("Your Question:")
+
+if st.button("Get Responses"):
+    if user_question:
+        responses = generate_group_therapy_responses(user_question)
+        for response in responses:
+            st.write(response)
+    else:
+        st.write("Please enter a question.")