Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ st.title('PromptLab')
|
|
9 |
# Create two columns for the Shinobi and Raikage buttons
|
10 |
col1, col2 = st.columns(2)
|
11 |
|
12 |
-
mode = st.radio("Choose a mode:", ["
|
13 |
|
14 |
# Retrieve the API key from Streamlit secrets
|
15 |
GOOGLE_API_KEY = st.secrets["GEMINI_API_KEY"]
|
@@ -25,7 +25,7 @@ st.write(f"You selected: {mode}")
|
|
25 |
|
26 |
|
27 |
# Shinobi and Raikage templates
|
28 |
-
|
29 |
Analyze the following user prompt and transform it into an enhanced, structured prompt following these steps:
|
30 |
|
31 |
1. Determine the primary subject area and required expertise level
|
@@ -49,7 +49,7 @@ The enhanced prompt should provide clear guidance on both content and format, ma
|
|
49 |
|
50 |
"""
|
51 |
|
52 |
-
|
53 |
You are an elite-level [role] with deep expertise in [subject].
|
54 |
Your task is to develop a structured, high-quality response following these key elements:
|
55 |
|
@@ -79,9 +79,9 @@ if st.button("Generate Enhanced Prompt"):
|
|
79 |
with st.spinner("Enhancing your prompt..."):
|
80 |
# Choose the template based on the selected mode
|
81 |
if mode == "Shinobi":
|
82 |
-
prompt =
|
83 |
else:
|
84 |
-
prompt =
|
85 |
|
86 |
# Initialize the generative model
|
87 |
model = genai.GenerativeModel('gemini-2.0-flash')
|
|
|
9 |
# Create two columns for the Shinobi and Raikage buttons
|
10 |
col1, col2 = st.columns(2)
|
11 |
|
12 |
+
mode = st.radio("Choose a mode:", ["Proficient Level", "Apex Level"], horizontal=True)
|
13 |
|
14 |
# Retrieve the API key from Streamlit secrets
|
15 |
GOOGLE_API_KEY = st.secrets["GEMINI_API_KEY"]
|
|
|
25 |
|
26 |
|
27 |
# Shinobi and Raikage templates
|
28 |
+
Proficient_TEMPLATE = """
|
29 |
Analyze the following user prompt and transform it into an enhanced, structured prompt following these steps:
|
30 |
|
31 |
1. Determine the primary subject area and required expertise level
|
|
|
49 |
|
50 |
"""
|
51 |
|
52 |
+
Apex_TEMPLATE = """
|
53 |
You are an elite-level [role] with deep expertise in [subject].
|
54 |
Your task is to develop a structured, high-quality response following these key elements:
|
55 |
|
|
|
79 |
with st.spinner("Enhancing your prompt..."):
|
80 |
# Choose the template based on the selected mode
|
81 |
if mode == "Shinobi":
|
82 |
+
prompt = Proficient_TEMPLATE.format(user_prompt=topic)
|
83 |
else:
|
84 |
+
prompt = Apex_TEMPLATE.format(user_prompt=topic)
|
85 |
|
86 |
# Initialize the generative model
|
87 |
model = genai.GenerativeModel('gemini-2.0-flash')
|