File size: 1,396 Bytes
586f060
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
f566d54
586f060
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import streamlit as st

# Define Roles and their Descriptions
roles = {
    "Writing Expert": "📝 Exhibits expertise in generating textual content and narratives.",
    "Roleplay Expert": "🎭 Specialized in mimicking behaviors or characters.",
    "Extraction Expert": "🔍 Strictly sticks to facts and extracts concise information.",
    "Mathematician": "➗ Solves mathematical problems with precision.",
    "Coder": "💻 Creates short python code functions to solve tasks.",
    "Reasoning Expert": "🤔 Analyzes situations and provides logical solutions.",
    "STEM Expert": "🔬 Specialized in Science, Technology, Engineering, and Mathematics tasks.",
    "Humanities Expert": "📚 Focuses on arts, literature, history, and other humanities subjects.",
}

# Streamlit UI
st.title("AI Role Selector")

# Dropdown to select role
selected_role = st.selectbox("Select AI Role:", list(roles.keys()))

# Switch to choose between two models
model = st.radio("Choose Model:", ["model_1", "model_2"])

# Text area for user input
user_input = st.text_area("Provide your task/question:")

# Button to execute
if st.button("Execute"):
    # Here, you would add code to get the AI response based on the selected role and model.
    # For now, just echoing the user input.
    st.write(f"You said: {user_input}")

# Display the description of the selected role
st.write(roles[selected_role])