main.py

# from dotenv import load_dotenv
# load_dotenv()

import streamlit as st
from typing_extensions import override
from openai import AssistantEventHandler
from streamlit_extras.buy_me_a_coffee import button
button(username="simjoy", floating=True, width=221)

import streamlit.components.v1 as components

st.header("리뷰 제조기 🤖")
st.write("👀 나는 T라서 리뷰에 팩트만 말해, F감성 리뷰는 내가 만들어줄게,")
st.write("업종과 별점을 입력해줘🙏 (글자수도 제한할 수 있어!💪) 바로 생성해줄께 🍀")

from openai import OpenAI
client = OpenAI()
  
# assistant = client.beta.assistants.create(
#   name="Review Generator",
#   instructions="You are a review maker. Once you receive the industry and star rating, create a review according to that.",
#   tools=[{"type": "code_interpreter"}],
#   model="gpt-4o",
# )

thread = client.beta.threads.create()

if 'messages' not in st.session_state:
    st.session_state.messages = []
 
class EventHandler(AssistantEventHandler):    
  @override
  def on_text_created(self, text) -> None:
    print(f"\nassistant > ", end="", flush=True)
      
  @override
  def on_text_delta(self, delta, snapshot):
    print(delta.value, end="", flush=True)
    st.session_state.messages.append(delta.value)
      
  def on_tool_call_created(self, tool_call):
    print(f"\nassistant > {tool_call.type}\n", flush=True)
  
  def on_tool_call_delta(self, delta, snapshot):
    if delta.type == 'code_interpreter':
      if delta.code_interpreter.input:
        print(f"delta.code : {delta.code_interpreter.input}", end="", flush=True)
      if delta.code_interpreter.outputs:
        print(f"\n\n output >", flush=True)
        for output in delta.code_interpreter.outputs:
          if output.type == "logs":
            print(f"\n logs : {output.logs}", flush=True)

with st.form("my_form"):
    col1, col2 = st.columns(2)
    with col1:
      upjong = st.text_input("업종을 입력하세요.")
      st.text("별점을 입력하세요.")
      sentiment_mapping = ["1", "2", "3", "4", "5"]
      selected = st.feedback("stars")
    with col2:
      number = st.number_input(
          "글자수 제한", value=200, step=100, placeholder="리뷰의 길이를 입력하세요."
      )
    
    submitted = st.form_submit_button("리뷰 생성")
    if selected is not None:
      message = client.beta.threads.messages.create(
          thread_id=thread.id,
          role="user",
          content="업종: " + upjong + ", 별점: " + sentiment_mapping[selected] + "점 이라고 입력했습니다. 리뷰 글자수는 "+str(number)+" 자리까지 리뷰를 생성해주세요. 그리고 이모지도 넣어서 리뷰를 마무리해주세요."
      )
      print(message)
    if submitted:
        st.session_state.messages = []
        with st.spinner('리뷰 생성중 🥳'):
          with client.beta.threads.runs.stream(
              thread_id=thread.id,
              assistant_id="asst_0etURnvod9b5nkDCr4SNIssc",
              event_handler=EventHandler(),
          ) as stream:
              stream.until_done()
          
          st.write_stream(st.session_state.messages)


# AdSense code
adsense_code = """
<script async src="https://pagead2.googlesyndication.com/pagead/js/adsbygoogle.js?client=ca-pub-2532708487251314"
     crossorigin="anonymous"></script>
<ins class="adsbygoogle"
     style="display:block; text-align:center;"
     data-ad-layout="in-article"
     data-ad-format="fluid"
     data-ad-client="ca-pub-2532708487251314"
     data-ad-slot="5381863461"></ins>
<script>
     (adsbygoogle = window.adsbygoogle || []).push({});
</script>
"""

components.html(adsense_code, width=200, height=200)