Spaces:
Sleeping
Sleeping
import streamlit as st | |
import os | |
from dotenv import load_dotenv | |
from langchain_community.llms import OpenAI | |
from langchain_google_genai import ChatGoogleGenerativeAI | |
# Load environment variables | |
load_dotenv() | |
def get_project_guide(project): | |
prompt = f""" | |
Break down the following project into simple, easy-to-follow tasks. Provide clear instructions for each step. | |
Project: {project} | |
Here is a detailed step-by-step guide: | |
""" | |
model = "Open AI" | |
if model == "Open AI": | |
llm = OpenAI(temperature=0.7, openai_api_key=st.secrets["OPENAI_API_KEY"]) | |
guide = llm(prompt) | |
elif model == "Gemini": | |
llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=st.secrets["GOOGLE_API_KEY"]) | |
guide = llm.invoke(prompt) | |
guide = guide.content | |
return guide | |
def app(): | |
st.title("Project Guide") | |
if 'projects' in st.session_state: | |
projects = st.session_state['projects'] | |
selected_project = st.selectbox("Select a project to get a detailed guide", projects) | |
if st.button("Get Project Guide"): | |
if selected_project: | |
guide = get_project_guide(selected_project) | |
st.write("Project Guide:") | |
st.text(guide) | |
else: | |
st.error("Please select a project.") | |
else: | |
st.error("Please go to the Project Suggestions page first to get a list of projects.") | |