AITechMentor / src /streamlit_app.py
sylaork's picture
Update src/streamlit_app.py
0254e0c verified
from openai import OpenAI
import streamlit as st
client = OpenAI(api_key="your_api_key")
def generate_learning_plan(goal, level):
prompt = f"Create a detailed, step by step from the scratch and structured 1-week learning plan for someone at {level} level who wants to learn {goal}."
response = client.chat.completions.create(
model="gpt-3.5-turbo",
messages=[{"role": "user", "content": prompt}]
)
return response['choices'][0]['message']['content']
def main():
st.title('AI Tech Mentor')
st.write('If you are struggling with finding your very own carrer path and looking for a way to start your journey, this is where you are gonna make your first step.')
st.write('Please enter your expected info and let us cook!')
name=st.text_input('Please enter your name:')
if name=='':
st.warning('Please enter your name so that we can know how to help you better.')
else:
st.write('Hi ' +name+ ' ! Welcome to our AI Tech Mentor. We are here to help you find your own carrer path. Let us start by asking some questions.')
goal=st.text_input('Please enter your goal:')
if goal=='':
st.warning('Please enter your goal so that we can know how to help you better.')
else:
st.write('Great! That such an exciting goal!')
st.write('What is your current knowledge level in software development?')
level=st.selectbox('What is your current knowledge level in software development?', ['I have no knowledge in software development', 'I did 5 or 10 mini projects', 'I have been working on projects for 1-2 years','Advanced'])
if st.button("Create Plan"):
if goal.strip() == "":
st.warning("Please enter")
return
with st.spinner("Creating the plan...."):
plan = generate_learning_plan(goal, level)
st.subheader(f'\n🧠 {name} , we have created a learning plan for you based on your goal and level of knowledge. Here is what we recommend:\n{plan}')
st.write(plan)
if __name__ == "__main__":
main()