networks:
  default:
    external:
      name: my_network

services:
  mensa:
    image: python
    volumes:
      - /mnt/docker/mensa:/app
    working_dir: /app
    entrypoint: python -m http.server

  yauk-net:
    image: docker.yauk.tv/streamlit
    restart: unless-stopped
    entrypoint: bash -c
    command:
      - |
        pip install redis openai
        cat <<"EOF" > streamlit_app.py
        import streamlit as st
        st.set_page_config(page_title="오늘놀자", page_icon=":kr:", layout='wide')
        import os, re, logging, uuid, json, time, redis, openai

        def render_equation(response):
            response = re.sub(r'\\\[([^\]]+?)\\\]', r'$$\1$$', response)
            response = re.sub(r'\\\(([^\)]+?)\\\)', r'$\1$', response)
            return response


        r = redis.Redis('redis', db=1)
        st.html("""<style>
        @import url("https://cdn.yauk.net/pretendard/dist/web/variable/pretendardvariable-dynamic-subset.css");

        header, ::-webkit-scrollbar { display: none !important; }

        *:not(pre):not(code):not(kbd):not(.katex):not(span) {
            font-family: -apple-system, "Pretendard Variable", sans-serif !important;
            font-weight: 300;
            strong { font-weight: 500; }
        }

        code {
            font-family: "Source Code Pro", "Pretendard Variable", monospace;
            color: #37e0b5;
        }

        pre * {
            font-size: 12px !important;
        }

        h1, h2, h3, h4, h5, h6 { font-weight: 250 !important; }


        .block-container {
            /*max-width:46rem; */
            padding-left: 1rem;
            padding-right: 2rem;
            @media (orientation: landscape) { padding-top: 1rem; }
        }
        </style>""")


        logging.basicConfig(
            level=logging.INFO, format='%(asctime)s - %(message)s',
            handlers=[logging.FileHandler("app.log"), ]
        )


        client = openai.OpenAI(api_key="$OPENAI_API_KEY")

        # visible page starts

        st.subheader('🕊️ 오늘놀자', divider='gray')

        # print history

        if "messages" not in st.session_state:
            st.session_state.messages = []
            st.session_state["session_id"] = str(uuid.uuid4())

        for message in st.session_state.messages:
            with st.chat_message(message["role"]):
                st.markdown(render_equation(message["content"]))

        # user prompt
        if prompt := st.chat_input("Message ChatGTP"):
            with st.chat_message("user"):
                st.write(prompt)
                st.session_state.messages.append({"role": "user", "content": prompt})
                logging.info(f"{st.session_state['session_id']} - User: {prompt}")


            # generate reseponse
            with st.chat_message("assistant"):
                stream = client.chat.completions.create(
                    model="gpt-4o-mini",
                    messages=[{"role": msg["role"], "content": msg["content"]}
                        for msg in st.session_state.messages],
                    temperature=0,
                    stream=True,
                )

                response, buffer = "", ""
                def wrapper():
                    global response, buffer
                    for chunk in stream:
                        chunk = chunk.choices[0].delta.content
                        if chunk is not None:
                            response += chunk
                            buffer += chunk
                            if buffer.count("\\[") == buffer.count("\\]") and \
                                buffer.count("\\(") == buffer.count("\\)") and \
                                not buffer.endswith("\\"):
                                    yield render_equation(buffer)
                                    buffer = ""
                with st.empty():
                    st.write_stream(wrapper())
                    if buffer:
                        st.write(render_equation(response))


                logging.info(f"{st.session_state['session_id']} - Model: {response}")
                st.session_state.messages.append({"role": "assistant", "content": response})

                r.rpush('yauk.net', json.dumps({
                    'timestamp': time.time(),
                    'session_id': st.session_state['session_id'],
                    'messages': st.session_state.messages,
                }))
        EOF
        streamlit run streamlit_app.py