Lyzr SDK를 사용하여 생산성 보조 도구 구축

import streamlit as st
from lyzr_automata.ai_models.openai import OpenAIModel
from lyzr_automata import Agent, Task
from lyzr_automata.pipelines.linear_sync_pipeline import LinearSyncPipeline
from PIL import Image
from lyzr_automata.tasks.task_literals import InputType, OutputType
import os

os.environ["OPENAI_API_KEY"] = st.secrets["apikey"]

st.title("Productivity Assistant?‍?")
st.markdown("Welcome to Productivity Assistant! We provide personalized tips and actionable advice to help you overcome challenges and achieve your specific goals efficiently.")
st.markdown("1) Mention your daily routine.")
st.markdown("2) Mention the productivity challenges you face.")
st.markdown("3) Mention your goals (Short Term or Long Term) or any other milestones you want to achieve if any.")
input = st.text_input("Please enter the above details:", placeholder="Type here")

open_ai_text_completion_model = OpenAIModel(
    api_key=st.secrets["apikey"],
    parameters={
        "model": "gpt-4-turbo-preview",
        "temperature": 0.2,
        "max_tokens": 1500,
    },
)

def generation(input):
    generator_agent = Agent(
        role="Expert PRODUCTIVITY ASSISTANT",
        prompt_persona="Your task is to offer PERSONALIZED PRODUCTIVITY TIPS and ACTIONABLE RECOMMENDATIONS tailored to an individual's DAILY ROUTINE, the PRODUCTIVITY CHALLENGES they encounter, and their GOALS—whether SHORT-TERM or LONG-TERM—or any other MILESTONES they aim to achieve.")
    prompt = """
[prompts here]
"""
    generator_agent_task = Task(
        name="Generation",
        model=open_ai_text_completion_model,
        agent=generator_agent,
        instructions=prompt,
        default_input=input,
        output_type=OutputType.TEXT,
        input_type=InputType.TEXT,
    ).execute()
    return generator_agent_task

if st.button("Assist!"):
    solution = generation(input)
    st.markdown(solution)