initial version
This commit is contained in:
Khanh Dinh
commit
bc2e200a2b
5
.idea/.gitignore
generated
vendored
Normal file
5
.idea/.gitignore
generated
vendored
Normal file
@ -0,0 +1,5 @@
|
||||
__pycache__/
|
||||
*.pyc
|
||||
.vscode/
|
||||
.env
|
||||
venv/
|
||||
10
.idea/OKR_Proposer.iml
generated
Normal file
10
.idea/OKR_Proposer.iml
generated
Normal file
@ -0,0 +1,10 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<module type="PYTHON_MODULE" version="4">
|
||||
<component name="NewModuleRootManager">
|
||||
<content url="file://$MODULE_DIR$">
|
||||
<excludeFolder url="file://$MODULE_DIR$/venv" />
|
||||
</content>
|
||||
<orderEntry type="inheritedJdk" />
|
||||
<orderEntry type="sourceFolder" forTests="false" />
|
||||
</component>
|
||||
</module>
|
||||
6
.idea/inspectionProfiles/profiles_settings.xml
generated
Normal file
6
.idea/inspectionProfiles/profiles_settings.xml
generated
Normal file
@ -0,0 +1,6 @@
|
||||
<component name="InspectionProjectProfileManager">
|
||||
<settings>
|
||||
<option name="USE_PROJECT_PROFILE" value="false" />
|
||||
<version value="1.0" />
|
||||
</settings>
|
||||
</component>
|
||||
4
.idea/misc.xml
generated
Normal file
4
.idea/misc.xml
generated
Normal file
@ -0,0 +1,4 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project version="4">
|
||||
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.9 (OKR_Proposer)" project-jdk-type="Python SDK" />
|
||||
</project>
|
||||
8
.idea/modules.xml
generated
Normal file
8
.idea/modules.xml
generated
Normal file
@ -0,0 +1,8 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project version="4">
|
||||
<component name="ProjectModuleManager">
|
||||
<modules>
|
||||
<module fileurl="file://$PROJECT_DIR$/.idea/OKR_Proposer.iml" filepath="$PROJECT_DIR$/.idea/OKR_Proposer.iml" />
|
||||
</modules>
|
||||
</component>
|
||||
</project>
|
||||
69
.idea/workspace.xml
generated
Normal file
69
.idea/workspace.xml
generated
Normal file
@ -0,0 +1,69 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project version="4">
|
||||
<component name="ChangeListManager">
|
||||
<list default="true" id="318c72a5-e9ec-4dd0-ac1a-cb0b5e2a86c2" name="Changes" comment="" />
|
||||
<option name="SHOW_DIALOG" value="false" />
|
||||
<option name="HIGHLIGHT_CONFLICTS" value="true" />
|
||||
<option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" />
|
||||
<option name="LAST_RESOLUTION" value="IGNORE" />
|
||||
</component>
|
||||
<component name="FileTemplateManagerImpl">
|
||||
<option name="RECENT_TEMPLATES">
|
||||
<list>
|
||||
<option value="Python Script" />
|
||||
</list>
|
||||
</option>
|
||||
</component>
|
||||
<component name="MarkdownSettingsMigration">
|
||||
<option name="stateVersion" value="1" />
|
||||
</component>
|
||||
<component name="ProjectId" id="2qwhdtwwl3dnedx4FlvRdDYZobj" />
|
||||
<component name="ProjectViewState">
|
||||
<option name="hideEmptyMiddlePackages" value="true" />
|
||||
<option name="showLibraryContents" value="true" />
|
||||
</component>
|
||||
<component name="PropertiesComponent">
|
||||
<property name="RunOnceActivity.OpenProjectViewOnStart" value="true" />
|
||||
<property name="RunOnceActivity.ShowReadmeOnStart" value="true" />
|
||||
<property name="last_opened_file_path" value="$PROJECT_DIR$" />
|
||||
</component>
|
||||
<component name="RecentsManager">
|
||||
<key name="CopyFile.RECENT_KEYS">
|
||||
<recent name="$PROJECT_DIR$" />
|
||||
</key>
|
||||
</component>
|
||||
<component name="RunManager">
|
||||
<configuration name="main" type="PythonConfigurationType" factoryName="Python" nameIsGenerated="true">
|
||||
<module name="OKR_Proposer" />
|
||||
<option name="INTERPRETER_OPTIONS" value="" />
|
||||
<option name="PARENT_ENVS" value="true" />
|
||||
<envs>
|
||||
<env name="PYTHONUNBUFFERED" value="1" />
|
||||
</envs>
|
||||
<option name="SDK_HOME" value="" />
|
||||
<option name="WORKING_DIRECTORY" value="$PROJECT_DIR$" />
|
||||
<option name="IS_MODULE_SDK" value="true" />
|
||||
<option name="ADD_CONTENT_ROOTS" value="true" />
|
||||
<option name="ADD_SOURCE_ROOTS" value="true" />
|
||||
<option name="SCRIPT_NAME" value="$PROJECT_DIR$/main.py" />
|
||||
<option name="PARAMETERS" value="" />
|
||||
<option name="SHOW_COMMAND_LINE" value="false" />
|
||||
<option name="EMULATE_TERMINAL" value="false" />
|
||||
<option name="MODULE_MODE" value="false" />
|
||||
<option name="REDIRECT_INPUT" value="false" />
|
||||
<option name="INPUT_FILE" value="" />
|
||||
<method v="2" />
|
||||
</configuration>
|
||||
</component>
|
||||
<component name="SpellCheckerSettings" RuntimeDictionaries="0" Folders="0" CustomDictionaries="0" DefaultDictionary="application-level" UseSingleDictionary="true" transferred="true" />
|
||||
<component name="TaskManager">
|
||||
<task active="true" id="Default" summary="Default task">
|
||||
<changelist id="318c72a5-e9ec-4dd0-ac1a-cb0b5e2a86c2" name="Changes" comment="" />
|
||||
<created>1735579636156</created>
|
||||
<option name="number" value="Default" />
|
||||
<option name="presentableId" value="Default" />
|
||||
<updated>1735579636156</updated>
|
||||
</task>
|
||||
<servers />
|
||||
</component>
|
||||
</project>
|
||||
7
.vscode/launch.json
vendored
Normal file
7
.vscode/launch.json
vendored
Normal file
@ -0,0 +1,7 @@
|
||||
{
|
||||
// Use IntelliSense to learn about possible attributes.
|
||||
// Hover to view descriptions of existing attributes.
|
||||
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
|
||||
"version": "0.2.0",
|
||||
"configurations": []
|
||||
}
|
||||
39
api.py
Normal file
39
api.py
Normal file
@ -0,0 +1,39 @@
|
||||
import requests
|
||||
import os
|
||||
import streamlit as st
|
||||
from dotenv import load_dotenv
|
||||
from settings import load_settings
|
||||
from utils import construct_prompt
|
||||
|
||||
# Load API key from .env file
|
||||
load_dotenv()
|
||||
api_key = os.getenv("OPENAI_API_KEY")
|
||||
|
||||
if not api_key:
|
||||
st.error("API key not found. Please set OPENAI_API_KEY in your .env file.")
|
||||
st.stop()
|
||||
|
||||
api_url = "https://genai.dev.odp.lhgroup.de/openai/deployments/gpt-4-turbo/chat/completions?api-version=2023-07-01-preview"
|
||||
|
||||
def fetch_okrs(user_input: str):
|
||||
settings = load_settings()
|
||||
|
||||
system_prompt = settings["system_prompt"]
|
||||
input_template = settings["input_template"]
|
||||
|
||||
user_prompt = construct_prompt(prompt_template=input_template, user_input=user_input)
|
||||
|
||||
headers = {"api-key": api_key, "Content-Type": "application/json"}
|
||||
body = {
|
||||
"messages": [
|
||||
{"role": "system", "content": system_prompt},
|
||||
{"role": "user", "content": user_prompt}
|
||||
]
|
||||
}
|
||||
try:
|
||||
response = requests.post(url=api_url, headers=headers, json=body)
|
||||
response.raise_for_status()
|
||||
return response.json()
|
||||
except Exception as e:
|
||||
st.error(f"Error fetching data from API: {e}")
|
||||
return None
|
||||
135
app.py
Normal file
135
app.py
Normal file
@ -0,0 +1,135 @@
|
||||
import streamlit as st
|
||||
import os
|
||||
import requests
|
||||
import json
|
||||
|
||||
|
||||
from styles import apply_styles
|
||||
|
||||
from streamlit_option_menu import option_menu
|
||||
|
||||
# Import other pages or modules
|
||||
from settings import settings_page, load_settings
|
||||
from proposer import proposer_page
|
||||
|
||||
apply_styles()
|
||||
#proposer_page()
|
||||
|
||||
|
||||
# Navigation Menu
|
||||
selected = option_menu(
|
||||
menu_title="Main Menu",
|
||||
options=["Home", "Settings"],
|
||||
icons=["house", "gear"],
|
||||
menu_icon="cast",
|
||||
default_index=0,
|
||||
orientation="horizontal",
|
||||
)
|
||||
|
||||
# Render pages based on selection
|
||||
if selected == "Home":
|
||||
# Home page logic goes here (e.g., OKR generator)
|
||||
#from utils import construct_prompt # Example import; replace with actual logic.
|
||||
|
||||
proposer_page()
|
||||
|
||||
elif selected == "Settings":
|
||||
settings_page()
|
||||
|
||||
|
||||
'''
|
||||
|
||||
# Streamlit App Layout
|
||||
st.title("OKR Generator")
|
||||
|
||||
# Input Section and Buttons Row
|
||||
st.subheader("Enter your idea or goal:")
|
||||
user_input = st.text_area(
|
||||
"Input your idea here:",
|
||||
value=st.session_state.get("user_input", INPUT_TEMPLATE.strip()),
|
||||
height=200,
|
||||
)
|
||||
|
||||
col1, col2 = st.columns([1, 1])
|
||||
with col1:
|
||||
if st.button("Reset All"):
|
||||
st.session_state.clear()
|
||||
with col2:
|
||||
generate_okrs_clicked = st.button("Generate OKRs")
|
||||
|
||||
if generate_okrs_clicked:
|
||||
if not user_input.strip():
|
||||
st.warning("Please provide some input before generating OKRs.")
|
||||
else:
|
||||
with st.spinner("Generating OKRs..."):
|
||||
# Construct prompt and call API
|
||||
prompt = construct_prompt(prompt_template=PROMPT_TEMPLATE, user_input=user_input)
|
||||
response = fetch_okrs(user_prompt=prompt, system_prompt=SYSTEM_PROMPT)
|
||||
if response:
|
||||
# Extract Objective and Key Results from response
|
||||
objective, key_results = extract_llm_response(response)
|
||||
st.session_state["objective"] = objective
|
||||
st.session_state["key_results"] = key_results
|
||||
|
||||
# Display Results Only if an OKR Has Been Generated
|
||||
if "objective" in st.session_state and "key_results" in st.session_state:
|
||||
# Display Objective Field with Responsibles Input Below It
|
||||
st.subheader("Proposal Objective:")
|
||||
objective_text = st.text_area(
|
||||
"Proposal Objective:",
|
||||
value=st.session_state.get("objective", ""),
|
||||
height=100,
|
||||
)
|
||||
|
||||
responsible_for_objective = st.text_input(
|
||||
"Responsibles for Objective (comma-separated):",
|
||||
value="",
|
||||
placeholder="e.g., Khanh Dinh, John Doe",
|
||||
key="responsibles_objective"
|
||||
)
|
||||
|
||||
# Display Key Results with Responsibles Below Each One
|
||||
st.subheader("Proposal Key Results:")
|
||||
key_result_boxes = []
|
||||
responsibles_for_key_results = []
|
||||
|
||||
for i, kr in enumerate(st.session_state["key_results"], start=1):
|
||||
kr_text = st.text_area(f"Key Result {i}:", value=kr, key=f"kr_{i}")
|
||||
responsible_for_kr = st.text_input(
|
||||
f"Responsibles for Key Result {i} (comma-separated):",
|
||||
value="",
|
||||
placeholder="e.g., Khanh Dinh",
|
||||
key=f"responsibles_kr_{i}"
|
||||
)
|
||||
|
||||
key_result_boxes.append(kr_text)
|
||||
responsibles_for_key_results.append(responsible_for_kr)
|
||||
|
||||
# Finalize Button Center-Aligned
|
||||
#finalize_col = st.columns([3, 2, 3])[1]
|
||||
#with finalize_col:
|
||||
if st.button("Finalize"):
|
||||
finalized_objective = objective_text.strip()
|
||||
finalized_key_results = [st.session_state[f"kr_{i+1}"].strip() for i in range(len(key_result_boxes))]
|
||||
|
||||
# Append initials of responsibles to Objective and Key Results
|
||||
responsibles_list_objective = [name.strip() for name in responsible_for_objective.split(",") if name.strip()]
|
||||
initials_objective = [f"[{''.join([part[0] for part in name.split()]).upper()}]" for name in responsibles_list_objective]
|
||||
initials_str_objective = ", ".join(initials_objective)
|
||||
|
||||
finalized_objective += f" [{initials_str_objective}]"
|
||||
|
||||
finalized_key_results_with_initials = []
|
||||
for i, kr in enumerate(finalized_key_results):
|
||||
responsibles_list_kr = [name.strip() for name in responsibles_for_key_results[i].split(",") if name.strip()]
|
||||
initials_kr = [f"{''.join([part[0] for part in name.split()]).upper()}" for name in responsibles_list_kr]
|
||||
initials_str_kr = ", ".join(initials_kr)
|
||||
finalized_key_results_with_initials.append(f"KR{i+1}: [{initials_str_kr}] {kr}")
|
||||
|
||||
# Display finalized data in non-editable format (full width)
|
||||
st.subheader("Finalized Objective:")
|
||||
st.write(finalized_objective)
|
||||
|
||||
st.subheader("Finalized Key Results:")
|
||||
for kr in finalized_key_results_with_initials:
|
||||
st.write(kr)'''
|
||||
58
config.py
Normal file
58
config.py
Normal file
@ -0,0 +1,58 @@
|
||||
SYSTEM_PROMPT = """
|
||||
You are an OKR coach and support us, as beginners in OKRs, with the correct phrasing of OKRs.
|
||||
Always one proposal and provide the proposal in the requested format (json).
|
||||
If you think the input is not clear enough and the OKR could be improved, then provide a hint,
|
||||
what could be added to the user input in order to provide improved OKRs. The hint should also
|
||||
always provided within the json.
|
||||
|
||||
Format could look like this:
|
||||
{
|
||||
"hint": "any additional questions, which can be added to the user input to rerun the prompting",
|
||||
"proposals": [
|
||||
{
|
||||
"objective":"objective variant 1",
|
||||
"key_results": ["key result 1"]
|
||||
},
|
||||
...
|
||||
]
|
||||
}
|
||||
"""
|
||||
|
||||
INPUT_TEMPLATE = """
|
||||
We want to improve our SLA framework by finalizing the current SLA template, which is used for the negotiations with our ground service partners.
|
||||
The resulting SLA will be used in order to measure the quality of the service partner and issue penalties if targets are not met.
|
||||
We would like to improve this, to have a better steering function, e.g. by adding a bonus component to the SLA framework.
|
||||
|
||||
To achieve this, we need to develop a bonus concept which is viable, feasible and desirable.
|
||||
We need to simulate the concept, so that we don't risk issuing too much bonus which we cannot cover.
|
||||
We need to define organizational processes and responsibilities so that we are able to pay a bonus.
|
||||
|
||||
How should the OKR look like for the next cycle which starts in Jan 2025 and ends in Apr 2025?
|
||||
"""
|
||||
|
||||
# Prompt template for user input
|
||||
PROMPT_TEMPLATE = """
|
||||
Please help us in defining proper OKRs.
|
||||
Here is what we have thought about and we would like to phrase an OKR with up to 5 key results.
|
||||
|
||||
this is the user input:
|
||||
{user_input}
|
||||
|
||||
Please provide the response in json format.
|
||||
the three proposals and potentially further questions to the user to improve the OKR, should be structured in a json as response.
|
||||
Objective (key: objective)
|
||||
key results (list with the key results)
|
||||
The json could be structured like this:
|
||||
|
||||
{
|
||||
"hint": "any additional questions, which can be added to the user input to rerun the prompting",
|
||||
"proposals": [
|
||||
{
|
||||
"objective":"objective variant 1",
|
||||
"key_result_1":"key result 1",
|
||||
...
|
||||
},
|
||||
...
|
||||
]
|
||||
}
|
||||
"""
|
||||
42
main.py
Normal file
42
main.py
Normal file
@ -0,0 +1,42 @@
|
||||
import streamlit as st
|
||||
from dotenv import load_dotenv
|
||||
from api import fetch_okrs
|
||||
from utils import construct_prompt, extract_llm_response
|
||||
from styles import apply_styles
|
||||
from config import SYSTEM_PROMPT, INPUT_TEMPLATE
|
||||
|
||||
# Load environment variables
|
||||
load_dotenv()
|
||||
|
||||
# Apply custom styles
|
||||
apply_styles()
|
||||
|
||||
# Streamlit App Layout
|
||||
st.title("OKR Generator")
|
||||
st.subheader("Enter your idea or goal:")
|
||||
|
||||
user_input = st.text_area(
|
||||
"Input your idea here:",
|
||||
value=st.session_state.get("user_input", INPUT_TEMPLATE.strip()),
|
||||
height=200,
|
||||
)
|
||||
|
||||
if st.button("Generate OKRs"):
|
||||
if not user_input.strip():
|
||||
st.warning("Please provide some input before generating OKRs.")
|
||||
else:
|
||||
with st.spinner("Generating OKRs..."):
|
||||
prompt = construct_prompt(user_input)
|
||||
response = fetch_okrs(prompt=prompt, system_prompt=SYSTEM_PROMPT)
|
||||
if response:
|
||||
objective, key_results = extract_llm_response(response)
|
||||
st.session_state["objective"] = objective
|
||||
st.session_state["key_results"] = key_results
|
||||
|
||||
if "objective" in st.session_state:
|
||||
st.subheader("Proposal Objective:")
|
||||
st.write(st.session_state["objective"])
|
||||
|
||||
st.subheader("Proposal Key Results:")
|
||||
for kr in st.session_state["key_results"]:
|
||||
st.write(kr)
|
||||
116
proposer.py
Normal file
116
proposer.py
Normal file
@ -0,0 +1,116 @@
|
||||
import streamlit as st
|
||||
|
||||
from api import fetch_okrs
|
||||
from config import SYSTEM_PROMPT, INPUT_TEMPLATE, PROMPT_TEMPLATE
|
||||
from utils import construct_prompt, extract_llm_response
|
||||
|
||||
def proposer_page():
|
||||
# Streamlit App Layout
|
||||
st.title("OKR Generator")
|
||||
|
||||
# Input Section and Buttons Row
|
||||
st.subheader("Enter your idea or goal:")
|
||||
user_input = st.text_area(
|
||||
"Input your idea here:",
|
||||
value=st.session_state.get("user_input", INPUT_TEMPLATE.strip()),
|
||||
height=200,
|
||||
)
|
||||
|
||||
col1, col2 = st.columns([1, 1])
|
||||
with col1:
|
||||
if st.button("Reset All"):
|
||||
st.session_state.clear()
|
||||
with col2:
|
||||
generate_okrs_clicked = st.button("Generate OKRs")
|
||||
|
||||
if generate_okrs_clicked:
|
||||
if not user_input.strip():
|
||||
st.warning("Please provide some input before generating OKRs.")
|
||||
else:
|
||||
with st.spinner("Generating OKRs..."):
|
||||
# Construct prompt and call API
|
||||
#prompt = construct_prompt(prompt_template=PROMPT_TEMPLATE, user_input=user_input)
|
||||
response = fetch_okrs(user_input=user_input)
|
||||
if response:
|
||||
# Extract Objective and Key Results from response
|
||||
print(response)
|
||||
objective, key_results, hint = extract_llm_response(response)
|
||||
|
||||
st.session_state["objective"] = objective
|
||||
st.session_state["key_results"] = key_results
|
||||
st.session_state["hint"] = hint
|
||||
|
||||
#st.subheader("Hint to improve the OKR proposal")
|
||||
#st.text(hint)
|
||||
|
||||
# Display Results Only if an OKR Has Been Generated
|
||||
if "objective" in st.session_state and "key_results" in st.session_state:
|
||||
|
||||
# Display Objective Field with Responsibles Input Below It
|
||||
st.subheader("Proposal Objective:")
|
||||
objective_text = st.text_area(
|
||||
"Proposal Objective:",
|
||||
value=st.session_state.get("objective", ""),
|
||||
height=100,
|
||||
)
|
||||
|
||||
responsible_for_objective = st.text_input(
|
||||
"Responsibles for Objective (comma-separated):",
|
||||
value="",
|
||||
placeholder="e.g., Khanh Dinh, John Doe",
|
||||
key="responsibles_objective"
|
||||
)
|
||||
|
||||
# Display Key Results with Responsibles Below Each One
|
||||
st.subheader("Proposal Key Results:")
|
||||
key_result_boxes = []
|
||||
responsibles_for_key_results = []
|
||||
|
||||
for i, kr in enumerate(st.session_state["key_results"], start=1):
|
||||
kr_text = st.text_area(f"Key Result {i}:", value=kr, key=f"kr_{i}")
|
||||
responsible_for_kr = st.text_input(
|
||||
f"Responsibles for Key Result {i} (comma-separated):",
|
||||
value="",
|
||||
placeholder="e.g., Khanh Dinh",
|
||||
key=f"responsibles_kr_{i}"
|
||||
)
|
||||
|
||||
key_result_boxes.append(kr_text)
|
||||
responsibles_for_key_results.append(responsible_for_kr)
|
||||
|
||||
# Finalize Button Center-Aligned
|
||||
#finalize_col = st.columns([3, 2, 3])[1]
|
||||
#with finalize_col:
|
||||
if st.button("Finalize"):
|
||||
finalized_objective = objective_text.strip()
|
||||
finalized_key_results = [st.session_state[f"kr_{i+1}"].strip() for i in range(len(key_result_boxes))]
|
||||
|
||||
# Append initials of responsibles to Objective and Key Results
|
||||
responsibles_list_objective = [name.strip() for name in responsible_for_objective.split(",") if name.strip()]
|
||||
initials_objective = [f"[{''.join([part[0] for part in name.split()]).upper()}]" for name in responsibles_list_objective]
|
||||
initials_str_objective = ", ".join(initials_objective)
|
||||
|
||||
finalized_objective = f"{initials_str_objective} {finalized_objective}"
|
||||
|
||||
finalized_key_results_with_initials = []
|
||||
for i, kr in enumerate(finalized_key_results):
|
||||
responsibles_list_kr = [name.strip() for name in responsibles_for_key_results[i].split(",") if name.strip()]
|
||||
initials_kr = [f"{''.join([part[0] for part in name.split()]).upper()}" for name in responsibles_list_kr]
|
||||
initials_str_kr = ", ".join(initials_kr)
|
||||
finalized_key_results_with_initials.append(f"KR{i+1}: [{initials_str_kr}] {kr}")
|
||||
|
||||
# Display finalized data in non-editable format (full width)
|
||||
st.subheader("Finalized Objective:")
|
||||
st.code(
|
||||
body=finalized_objective,
|
||||
language=None,
|
||||
wrap_lines=True
|
||||
)
|
||||
|
||||
st.subheader("Finalized Key Results:")
|
||||
for kr in finalized_key_results_with_initials:
|
||||
st.code(
|
||||
body=kr,
|
||||
language=None,
|
||||
wrap_lines=True
|
||||
)
|
||||
74
requirements.txt
Normal file
74
requirements.txt
Normal file
@ -0,0 +1,74 @@
|
||||
altair==5.5.0
|
||||
appnope==0.1.4
|
||||
asttokens==3.0.0
|
||||
attrs==24.3.0
|
||||
blinker==1.9.0
|
||||
cachetools==5.5.0
|
||||
certifi==2024.12.14
|
||||
charset-normalizer==3.4.1
|
||||
click==8.1.8
|
||||
comm==0.2.2
|
||||
cramjam==2.9.1
|
||||
debugpy==1.8.11
|
||||
decorator==5.1.1
|
||||
et-xmlfile==2.0.0
|
||||
exceptiongroup==1.2.2
|
||||
executing==2.1.0
|
||||
fastparquet==2024.11.0
|
||||
fsspec==2024.12.0
|
||||
gitdb==4.0.11
|
||||
GitPython==3.1.43
|
||||
idna==3.10
|
||||
importlib-metadata==8.5.0
|
||||
ipykernel==6.29.5
|
||||
ipython==8.18.1
|
||||
jedi==0.19.2
|
||||
jinja2==3.1.5
|
||||
jsonschema==4.23.0
|
||||
jsonschema-specifications==2024.10.1
|
||||
jupyter-client==8.6.3
|
||||
jupyter-core==5.7.2
|
||||
markdown-it-py==3.0.0
|
||||
MarkupSafe==3.0.2
|
||||
matplotlib-inline==0.1.7
|
||||
mdurl==0.1.2
|
||||
narwhals==1.20.1
|
||||
nest-asyncio==1.6.0
|
||||
numpy==2.0.2
|
||||
openpyxl==3.1.5
|
||||
packaging==24.2
|
||||
pandas==2.2.3
|
||||
parso==0.8.4
|
||||
pexpect==4.9.0
|
||||
pillow==11.0.0
|
||||
platformdirs==4.3.6
|
||||
prompt-toolkit==3.0.48
|
||||
protobuf==5.29.2
|
||||
psutil==6.1.1
|
||||
ptyprocess==0.7.0
|
||||
pure-eval==0.2.3
|
||||
pyarrow==18.1.0
|
||||
pydeck==0.9.1
|
||||
pygments==2.18.0
|
||||
pypdf2==3.0.1
|
||||
python-dateutil==2.9.0.post0
|
||||
python-dotenv==1.0.1
|
||||
pytz==2024.2
|
||||
pyzmq==26.2.0
|
||||
referencing==0.35.1
|
||||
requests==2.32.3
|
||||
rich==13.9.4
|
||||
rpds-py==0.22.3
|
||||
six==1.17.0
|
||||
smmap==5.0.1
|
||||
stack-data==0.6.3
|
||||
streamlit==1.41.1
|
||||
tenacity==9.0.0
|
||||
toml==0.10.2
|
||||
tornado==6.4.2
|
||||
traitlets==5.14.3
|
||||
typing-extensions==4.12.2
|
||||
tzdata==2024.2
|
||||
urllib3==2.3.0
|
||||
wcwidth==0.2.13
|
||||
zipp==3.21.0
|
||||
202
settings.py
Normal file
202
settings.py
Normal file
@ -0,0 +1,202 @@
|
||||
import streamlit as st
|
||||
import json
|
||||
import os
|
||||
from config import SYSTEM_PROMPT, INPUT_TEMPLATE, PROMPT_TEMPLATE
|
||||
|
||||
def settings_page():
|
||||
st.title("Settings")
|
||||
|
||||
# Section for System Prompt
|
||||
st.subheader("System Prompt")
|
||||
if "system_prompt" not in st.session_state:
|
||||
st.session_state["system_prompt"] = SYSTEM_PROMPT.strip()
|
||||
system_prompt = st.text_area(
|
||||
"Define the system prompt:",
|
||||
value=st.session_state["system_prompt"]
|
||||
)
|
||||
|
||||
# Section for Input Template
|
||||
st.subheader("Input Template")
|
||||
if "input_template" not in st.session_state:
|
||||
st.session_state["input_template"] = INPUT_TEMPLATE.strip()
|
||||
input_template = st.text_area(
|
||||
"Define the input template:",
|
||||
value=st.session_state["input_template"],
|
||||
height=400
|
||||
)
|
||||
|
||||
# Section for Prompt Template
|
||||
st.subheader("Prompt Template")
|
||||
if "prompt_template" not in st.session_state:
|
||||
st.session_state["prompt_template"] = PROMPT_TEMPLATE.strip()
|
||||
prompt_template = st.text_area(
|
||||
"Define the prompt template:",
|
||||
value=st.session_state["prompt_template"]
|
||||
)
|
||||
|
||||
# Section for Number of Key Results
|
||||
st.subheader("Number of Key Results")
|
||||
if "num_key_results" not in st.session_state:
|
||||
st.session_state["num_key_results"] = 4
|
||||
num_key_results = st.number_input(
|
||||
"Set the maximum number of key results:",
|
||||
min_value=1,
|
||||
max_value=10,
|
||||
value=st.session_state["num_key_results"]
|
||||
)
|
||||
|
||||
# Save Button for Settings
|
||||
if st.button("Save Settings"):
|
||||
st.session_state["system_prompt"] = system_prompt
|
||||
st.session_state["input_template"] = input_template
|
||||
st.session_state["prompt_template"] = prompt_template
|
||||
st.session_state["num_key_results"] = num_key_results
|
||||
st.success("Settings saved successfully!")
|
||||
|
||||
# Section for Team Member Management
|
||||
team_member_management()
|
||||
|
||||
def team_member_management():
|
||||
st.subheader("Team Member Management")
|
||||
|
||||
# Initialize team members in session state
|
||||
if "team_members" not in st.session_state:
|
||||
st.session_state["team_members"] = ["John Doe", "Jane Smith"]
|
||||
|
||||
# Display current team members
|
||||
st.write("### Current Team Members")
|
||||
for member in st.session_state["team_members"]:
|
||||
col1, col2 = st.columns([4, 1])
|
||||
col1.write(member)
|
||||
if col2.button("Remove", key=f"remove_{member}"):
|
||||
st.session_state["team_members"].remove(member)
|
||||
st.experimental_rerun()
|
||||
|
||||
# Add new team member
|
||||
st.write("### Add New Team Member")
|
||||
new_member = st.text_input("Enter new team member name:", key="new_member")
|
||||
if st.button("Add Team Member"):
|
||||
if new_member.strip():
|
||||
if new_member.strip() not in st.session_state["team_members"]:
|
||||
st.session_state["team_members"].append(new_member.strip())
|
||||
st.success(f"Added {new_member.strip()} to the team.")
|
||||
st.experimental_rerun()
|
||||
else:
|
||||
st.warning(f"{new_member.strip()} is already in the team.")
|
||||
else:
|
||||
st.warning("Please enter a valid name.")
|
||||
|
||||
|
||||
|
||||
|
||||
SETTINGS_FILE = "settings.json"
|
||||
|
||||
# Load settings from JSON file
|
||||
def load_settings():
|
||||
if os.path.exists(SETTINGS_FILE) and 1 == 2:
|
||||
with open(SETTINGS_FILE, "r") as f:
|
||||
return json.load(f)
|
||||
else:
|
||||
# Default settings
|
||||
return {
|
||||
"system_prompt": SYSTEM_PROMPT,
|
||||
"input_template": INPUT_TEMPLATE,
|
||||
"prompt_template": PROMPT_TEMPLATE,
|
||||
"num_key_results": 4,
|
||||
"team_members": ["Khanh Dinh", "Robin Plitzko", "Roberto Renna"],
|
||||
"team_member_template": "{name} - Team Member"
|
||||
}
|
||||
|
||||
# Save settings to JSON file
|
||||
def save_settings(settings):
|
||||
with open(SETTINGS_FILE, "w") as f:
|
||||
json.dump(settings, f, indent=4)
|
||||
|
||||
# Settings Page
|
||||
def settings_page():
|
||||
st.title("Settings")
|
||||
|
||||
# Load settings
|
||||
settings = load_settings()
|
||||
|
||||
# Section for System Prompt
|
||||
st.subheader("System Prompt")
|
||||
system_prompt = st.text_area(
|
||||
"Define the system prompt:",
|
||||
value=settings["system_prompt"]
|
||||
)
|
||||
|
||||
# Section for Input Template
|
||||
st.subheader("Input Template")
|
||||
input_template = st.text_area(
|
||||
"Define the input template:",
|
||||
value=settings["input_template"]
|
||||
)
|
||||
|
||||
# Section for Prompt Template
|
||||
st.subheader("Prompt Template")
|
||||
prompt_template = st.text_area(
|
||||
"Define the prompt template (use {num_key_results} for dynamic insertion):",
|
||||
value=settings["prompt_template"]
|
||||
)
|
||||
|
||||
# Section for Number of Key Results
|
||||
st.subheader("Number of Key Results")
|
||||
num_key_results = st.number_input(
|
||||
"Set the maximum number of key results:",
|
||||
min_value=1,
|
||||
max_value=10,
|
||||
value=settings["num_key_results"]
|
||||
)
|
||||
|
||||
# Section for Team Member Template
|
||||
st.subheader("Team Member Template")
|
||||
team_member_template = st.text_input(
|
||||
"Define a template for team members (use {name} for dynamic insertion):",
|
||||
value=settings["team_member_template"]
|
||||
)
|
||||
|
||||
# Section for Team Member Management
|
||||
team_member_management(settings)
|
||||
|
||||
# Save Button for Settings
|
||||
if st.button("Save Settings"):
|
||||
# Update settings dictionary
|
||||
settings["system_prompt"] = system_prompt
|
||||
settings["input_template"] = input_template
|
||||
settings["prompt_template"] = prompt_template
|
||||
settings["num_key_results"] = num_key_results
|
||||
settings["team_member_template"] = team_member_template
|
||||
|
||||
# Save to file
|
||||
save_settings(settings)
|
||||
st.success("Settings saved successfully!")
|
||||
|
||||
# Team Member Management Section
|
||||
def team_member_management(settings):
|
||||
st.subheader("Team Member Management")
|
||||
|
||||
# Display current team members
|
||||
st.write("### Current Team Members")
|
||||
for member in settings["team_members"]:
|
||||
col1, col2 = st.columns([4, 1])
|
||||
col1.write(member)
|
||||
if col2.button(f"Remove {member}", key=f"remove_{member}"):
|
||||
settings["team_members"].remove(member)
|
||||
save_settings(settings)
|
||||
st.experimental_rerun()
|
||||
|
||||
# Add new team member
|
||||
st.write("### Add New Team Member")
|
||||
new_member = st.text_input("Enter new team member name:", key="new_member")
|
||||
if st.button("Add Team Member"):
|
||||
if new_member.strip():
|
||||
if new_member.strip() not in settings["team_members"]:
|
||||
settings["team_members"].append(new_member.strip())
|
||||
save_settings(settings)
|
||||
st.success(f"Added {new_member.strip()} to the team.")
|
||||
st.experimental_rerun()
|
||||
else:
|
||||
st.warning(f"{new_member.strip()} is already in the team.")
|
||||
else:
|
||||
st.warning("Please enter a valid name.")
|
||||
11
settings1.json
Normal file
11
settings1.json
Normal file
@ -0,0 +1,11 @@
|
||||
{
|
||||
"system_prompt": "You are an OKR coach and support us, as beginners in OKRs, with the correct phrasing of OKRs.",
|
||||
"input_template": "We want to improve our SLA framework by finalizing the current SLA template...",
|
||||
"prompt_template": "Please help us in defining proper OKRs. Here is what we have thought about and we would like to phrase an OKR with max. 4 key results. this is the user input: {user_input} Please provide the response in json format. Objective (key: objective) key results (list with the key results)",
|
||||
"num_key_results": 4,
|
||||
"team_members": [
|
||||
"John Doe",
|
||||
"Jane Smith"
|
||||
],
|
||||
"team_member_template": "{name} - Team Member"
|
||||
}
|
||||
12
styles.py
Normal file
12
styles.py
Normal file
@ -0,0 +1,12 @@
|
||||
import streamlit as st
|
||||
|
||||
def apply_styles():
|
||||
page_bg_color = """
|
||||
<style>
|
||||
body {
|
||||
background-color: #002F5F;
|
||||
color: white;
|
||||
}
|
||||
</style>
|
||||
"""
|
||||
st.markdown(page_bg_color, unsafe_allow_html=True)
|
||||
42
test.ipynb
Normal file
42
test.ipynb
Normal file
@ -0,0 +1,42 @@
|
||||
{
|
||||
"cells": [
|
||||
{
|
||||
"cell_type": "code",
|
||||
"execution_count": 1,
|
||||
"metadata": {},
|
||||
"outputs": [
|
||||
{
|
||||
"name": "stdout",
|
||||
"output_type": "stream",
|
||||
"text": [
|
||||
"hello\n"
|
||||
]
|
||||
}
|
||||
],
|
||||
"source": [
|
||||
"print('hello')"
|
||||
]
|
||||
}
|
||||
],
|
||||
"metadata": {
|
||||
"kernelspec": {
|
||||
"display_name": "venv",
|
||||
"language": "python",
|
||||
"name": "python3"
|
||||
},
|
||||
"language_info": {
|
||||
"codemirror_mode": {
|
||||
"name": "ipython",
|
||||
"version": 3
|
||||
},
|
||||
"file_extension": ".py",
|
||||
"mimetype": "text/x-python",
|
||||
"name": "python",
|
||||
"nbconvert_exporter": "python",
|
||||
"pygments_lexer": "ipython3",
|
||||
"version": "3.9.6"
|
||||
}
|
||||
},
|
||||
"nbformat": 4,
|
||||
"nbformat_minor": 2
|
||||
}
|
||||
160
utils.py
Normal file
160
utils.py
Normal file
@ -0,0 +1,160 @@
|
||||
import json
|
||||
import streamlit as st
|
||||
|
||||
# Function to construct the prompt
|
||||
def construct_prompt(prompt_template: str, user_input: str) -> str:
|
||||
return prompt_template.format(user_input=user_input)
|
||||
|
||||
'''# Function to extract and parse JSON response
|
||||
def extract_llm_response(response):
|
||||
print("response:", response)
|
||||
try:
|
||||
raw_message_content = response["choices"][0]["message"]["content"]
|
||||
print("raw_message_content:", raw_message_content)
|
||||
# Clean and parse the JSON content
|
||||
cleaned_content = raw_message_content.replace("`", "").split("json")[-1]
|
||||
|
||||
# for debugging
|
||||
#if debug:
|
||||
# print("cleaned:", '-'*50)
|
||||
# print(cleaned_content.strip())
|
||||
|
||||
def parse_json_content(cleaned_content: str):
|
||||
"""
|
||||
Parses the cleaned content to extract valid JSON data.
|
||||
|
||||
Args:
|
||||
cleaned_content (str): The raw content containing JSON data.
|
||||
|
||||
Returns:
|
||||
dict or list: The parsed JSON object.
|
||||
"""
|
||||
import re
|
||||
|
||||
# Step 1: Strip unwanted characters and clean the content
|
||||
cleaned_content = cleaned_content.strip()
|
||||
|
||||
# Step 2: Use regex to extract only the valid JSON block (e.g., starts with [ or {)
|
||||
json_match = re.search(r"(\{.*\}|\[.*\])", cleaned_content, re.DOTALL)
|
||||
|
||||
if not json_match:
|
||||
raise ValueError("No valid JSON found in the content.")
|
||||
|
||||
# Step 3: Extract and parse the valid JSON
|
||||
valid_json = json_match.group(0) # Extract matched JSON block
|
||||
try:
|
||||
extracted_data = json.loads(valid_json)
|
||||
except json.JSONDecodeError as e:
|
||||
raise ValueError(f"Failed to decode JSON. Error: {e}\nContent:\n{valid_json}")
|
||||
|
||||
return extracted_data
|
||||
|
||||
parsed_data = parse_json_content(cleaned_content=cleaned_content)
|
||||
print("parsed_data:",parsed_data)
|
||||
print("debug:", parsed_data.get("objective", ""))
|
||||
|
||||
#parsed_data = json.loads(cleaned_content)
|
||||
return parsed_data.get("objective", ""), parsed_data.get("key_results", [])
|
||||
except Exception as e:
|
||||
st.error(f"Error parsing API response: {e}")
|
||||
return "", []'''
|
||||
|
||||
import json
|
||||
import re
|
||||
|
||||
def parse_json_content(cleaned_content: str):
|
||||
"""
|
||||
Parses the cleaned content to extract valid JSON data.
|
||||
|
||||
Args:
|
||||
cleaned_content (str): The raw content containing JSON data.
|
||||
|
||||
Returns:
|
||||
dict or list: The parsed JSON object.
|
||||
"""
|
||||
import re
|
||||
|
||||
# Step 1: Strip unwanted characters and clean the content
|
||||
cleaned_content = cleaned_content.strip()
|
||||
|
||||
# Step 2: Use regex to extract only the valid JSON block (e.g., starts with [ or {)
|
||||
json_match = re.search(r"(\{.*\}|\[.*\])", cleaned_content, re.DOTALL)
|
||||
|
||||
if not json_match:
|
||||
raise ValueError("No valid JSON found in the content.")
|
||||
|
||||
# Step 3: Extract and parse the valid JSON
|
||||
valid_json = json_match.group(0) # Extract matched JSON block
|
||||
try:
|
||||
extracted_data = json.loads(valid_json)
|
||||
except json.JSONDecodeError as e:
|
||||
raise ValueError(f"Failed to decode JSON. Error: {e}\nContent:\n{valid_json}")
|
||||
|
||||
return extracted_data
|
||||
|
||||
# Function to extract and parse JSON response
|
||||
def extract_llm_response(response):
|
||||
"""
|
||||
Extracts and parses the JSON response from the API.
|
||||
|
||||
Args:
|
||||
response (dict): The API response containing a hint and proposals.
|
||||
|
||||
Returns:
|
||||
tuple: A tuple containing the objective (str), key results (list), and hint (str).
|
||||
"""
|
||||
print("RESPONSE:",response)
|
||||
|
||||
raw_message_content = response["choices"][0]["message"]["content"]
|
||||
print("raw_message_content:", raw_message_content)
|
||||
# Clean and parse the JSON content
|
||||
cleaned_content = raw_message_content.replace("`", "").split("json")[-1]
|
||||
print("cleaned content", cleaned_content)
|
||||
|
||||
parsed_data = parse_json_content(cleaned_content=cleaned_content)
|
||||
print("parsed_data:",parsed_data)
|
||||
|
||||
hint = parsed_data.get("hint", "")
|
||||
|
||||
proposals = parsed_data.get("proposals", [])
|
||||
|
||||
if proposals:
|
||||
# Extract the first proposal's objective and key results
|
||||
first_proposal = proposals[0] # Get the first proposal (assuming it's a list)
|
||||
objective = first_proposal.get("objective", "")
|
||||
key_results = first_proposal.get("key_results", [])
|
||||
else:
|
||||
objective = ""
|
||||
key_results = []
|
||||
|
||||
#print("debug:", parsed_data.get("objective", ""))
|
||||
|
||||
return objective, key_results, hint
|
||||
|
||||
#try:
|
||||
# Extract hint from the response
|
||||
hint = response.get("hint", "")
|
||||
|
||||
# Extract proposals from the response
|
||||
proposals = response.get("proposals", [])
|
||||
print("hint:", hint)
|
||||
print("proposals:", proposals)
|
||||
|
||||
# Check if proposals are available
|
||||
if proposals:
|
||||
# Extract the first proposal's objective and key results
|
||||
first_proposal = proposals[0] # Get the first proposal (assuming it's a list)
|
||||
objective = first_proposal.get("objective", "")
|
||||
key_results = first_proposal.get("key_results", [])
|
||||
else:
|
||||
objective = ""
|
||||
key_results = []
|
||||
|
||||
# Log parsed data for debugging
|
||||
print("parsed_data:", {"objective": objective, "key_results": key_results, "hint": hint})
|
||||
|
||||
return objective, key_results, hint
|
||||
|
||||
#except Exception as e:
|
||||
# print(f"Error parsing API response: {e}")
|
||||
# return "", [], ""
|
||||
Loading…
Reference in New Issue
Block a user