githubmirror/DemoGPT

src/alpha/app.py
128 строк3 KB
melihunsal
code reformatted
28 июл 2023, 03:27
28 июл 2023, 03:270022318
import logging
import os
import signal
import webbrowser
from time import sleep

import streamlit as st
import utils
from model import Model

# logging.basicConfig(level = logging.DEBUG,format='%(levelname)s-%(message)s')

try:
    from dotenv import load_dotenv

    load_dotenv()
except Exception as e:
    logging.error("dotenv import error but no needed")


def generate_response(txt):
    """
    Generate response using the LangChainCoder.

    Args:
        txt (str): The input text.

    Yields:
        dict: A dictionary containing response information.
    """
    for data in agent(txt):
        yield data


# Page title
title = "🧩 DemoGPT"

st.set_page_config(page_title=title)
st.title(title)
# Text input

openai_api_key = st.sidebar.text_input(
    "OpenAI API Key",
    placeholder="sk-...",
    value=os.getenv("OPENAI_API_KEY", ""),
    type="password",
)

model_name = st.sidebar.selectbox("Model", ("gpt-3.5-turbo", "gpt-4"))

empty_idea = st.empty()
demo_idea = empty_idea.text_area(
    "Enter your LLM-based demo idea", placeholder="Type your demo idea here", height=100
)


PROGRESS_BAR_INFO = {
    "start": {"text": "Plan generation started...", "percentage": 0},
    "plan": {"text": "Global plan has been generated", "percentage": 20},
    "explanation": {"text": "Tasks have been explained", "percentage": 40},
    "langchain": {"text": "Langchain code has been generated.", "percentage": 60},
    "streamlit": {"text": "Streamlit code has been generated...", "percentage": 80},
    "done": {"text": "App created, directed to the demo page", "percentage": 100},
}


def progressBarOld(key, bar=None):
    info = PROGRESS_BAR_INFO[key]
    if bar:
        bar.progress(info["percentage"], text=info["text"])
    else:
        return st.progress(info["percentage"], text=info["text"])


def progressBar(key, bar=None):
    info = PROGRESS_BAR_INFO[key]
    if bar:
        bar.progress(info["percentage"])
    else:
        return st.progress(info["percentage"])


if "pid" not in st.session_state:
    st.session_state["pid"] = -1


with st.form("a", clear_on_submit=True):
    submitted = st.form_submit_button("Submit")

if submitted:

    if not openai_api_key.startswith("sk-"):
        st.warning("Please enter your OpenAI API Key!", icon="⚠️")
    else:
        bar = progressBar("start")

        container = st.container()

        agent = Model(openai_api_key=openai_api_key)
        agent.setModel(model_name)

        if st.session_state["pid"] != -1:
            logging.info(f"Terminating the previous applicaton ...")
            os.kill(st.session_state["pid"], signal.SIGTERM)
            st.session_state["pid"] = -1

        code_empty = st.empty()
        for data in generate_response(demo_idea):
            stage = data["stage"]
            code = data.get("code")

            progressBar(stage, bar)

            if stage == "done":
                container.success(PROGRESS_BAR_INFO[stage]["text"])
                with st.expander("Code"):
                    st.code(code)
                example_submitted = False
                st.session_state["pid"] = utils.runStreamlit(code, openai_api_key)
                sleep(5)
                webbrowser.open("http://localhost:8502")
                break
            else:
                container.info("🤖 " + PROGRESS_BAR_INFO[stage]["text"])
                if stage == "plan":
                    container.json(data["tasks"])
                elif code:
                    container.code(code)