|
import os |
|
from groq import Groq |
|
import streamlit as st |
|
|
|
client = Groq( |
|
api_key=os.environ.get("GROQ_API_KEY"), |
|
) |
|
|
|
st.write('Background analysis') |
|
gl = st.text_area("Goal and assumptions", |
|
placeholder="""What is the assumption to verify, what do you want to know, |
|
the goal and mission of this polling""") |
|
bg = st.text_area("Background information", |
|
placeholder="""What is the polling about, what we should know to better |
|
help you design the questions""") |
|
|
|
if 'ft' in st.session_state or st.button('Next step'): |
|
if 'ft' not in st.session_state: |
|
rf = os.environ.get("DESIGN_PROMPT") |
|
messages = [ |
|
{ |
|
"role": "user", |
|
"content": f"""Generate possible factors related to the goal |
|
based on the information provided. List the factors in markdown list. |
|
You can give advice to human to edit their goal if you cannot define |
|
more than 3 factors.\n\n |
|
goals:\n\n{gl}\n\n |
|
background:\n\n{bg}\n\n |
|
some design principles for your reference:\n\n |
|
reference:\n\n```markdown\n{rf}\n\n```\n\n""", |
|
} |
|
] |
|
|
|
chat_completion = client.chat.completions.create( |
|
messages=messages, |
|
model="llama3-70b-8192", |
|
) |
|
st.session_state['ft'] = chat_completion.choices[0].message.content |
|
|
|
am = st.text_area("Assumption to be tested", |
|
placeholder="""What is the assumption you want to verify |
|
through the polling""") |
|
st.session_state['ft'] = st.text_area( |
|
"Factors to be considered in questions", |
|
st.session_state['ft'], |
|
height=300) |
|
st.info('Before next step. Edit above factors, add or remove if needed') |
|
|
|
ft = st.session_state['ft'] |
|
rf = os.environ.get("DESIGN_PROMPT") |
|
|
|
if st.button('Generate questions'): |
|
rf += f'''\n\n---\n\n{os.environ.get("DESIGN_PROMPT_SIXPOINT")}''' |
|
messages = [ |
|
{ |
|
"role": "user", |
|
"content": f"""Extract all factors from the text to create context. |
|
Generate 5 question based on the text and context.\n\n |
|
assumption to be tested through the questionnaire:\n\n{am}\n\n |
|
text:\n\n{ft}\n\n |
|
some design principles for your reference:\n\n |
|
reference:\n\n```markdown\n{rf}\n```\n\n""", |
|
} |
|
] |
|
|
|
chat_completion = client.chat.completions.create( |
|
messages=messages, |
|
model="llama3-70b-8192", |
|
) |
|
|
|
md = chat_completion.choices[0].message.content |
|
st.markdown(md) |
|
st.info('Suggested number of sampling is 5 to 10 times of number of questions', icon="ℹ️") |
|
|
|
|
|
if st.button('Restart'): |
|
del st.session_state['ft'] |
