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.")