Upload 2 files

app.py

@@ -0,0 +1,98 @@
+#    _____ _    _ _____    _____ _           _   _           _   
+#   / ____| |  | |_   _|  / ____| |         | | | |         | |  
+#  | |  __| |  | | | |   | |    | |__   __ _| |_| |__   ___ | |_ 
+#  | | |_ | |  | | | |   | |    | '_ \ / _` | __| '_ \ / _ \| __|
+#  | |__| | |__| |_| |_  | |____| | | | (_| | |_| |_) | (_) | |_ 
+#   \_____|\____/|_____|  \_____|_| |_|\__,_|\__|_.__/ \___/ \__|
+#                                                         
+#    __________  ___  ___________  _____   ___ 
+#   /  _/_  __/ / _ )/ __/_  __| \/ / _ | / _ \
+#  _/ /  / /   / _  / _/  / /   \  / __ |/ , _/
+# /___/ /_/   /____/___/ /_/    /_/_/ |_/_/|_| 
+#
+# SOFT PILLOW VERZIÓ
+# HF - Streamlit verzió
+
+
+# streamlit run softpillow1.4streamlitchat.py
+
+import streamlit as st
+import time
+from openai import OpenAI
+from itb_kulcs import api_key
+from itb_kulcs import assistant_id
+
+client = OpenAI(api_key=api_key) # Init OpenAI client
+
+ASSISTANT_ID = assistant_id # Assistant ID
+
+st.title("IT Betyár - Custom-ChatBot") # Streamlit App Focim
+
+# Initialize messages in session state if not present
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# Display existing messages
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.write(message["content"])
+
+# Chat input
+if prompt := st.chat_input("Type your message..."):
+    # Display user message
+    with st.chat_message("user"):
+        st.write(prompt)
+    
+    # Add user message to state
+    st.session_state.messages.append({"role": "user", "content": prompt})
+
+    # Get assistant response
+    with st.chat_message("assistant"):
+        with st.spinner("Thinking..."):
+            # Create thread and send message
+            thread = client.beta.threads.create()
+            client.beta.threads.messages.create(
+                thread_id=thread.id,
+                role="user",
+                content=prompt
+            )
+
+            # Run assistant
+            run = client.beta.threads.runs.create(
+                thread_id=thread.id,
+                assistant_id=ASSISTANT_ID
+            )
+
+            # Wait for completion
+            while run.status != "completed":
+                time.sleep(0.5)
+                run = client.beta.threads.runs.retrieve(
+                    thread_id=thread.id,
+                    run_id=run.id
+                )
+
+            # Get and display response
+            messages = client.beta.threads.messages.list(thread_id=thread.id)
+            response = messages.data[0].content[0].text.value
+            cleaned_response = response.split("【")[0]
+            st.write(cleaned_response)
+
+            # Add assistant response to state
+            st.session_state.messages.append(
+                {"role": "assistant", "content": cleaned_response}
+            )
+
+            # Clean up thread
+            client.beta.threads.delete(thread.id)
+
+
+# Teszt üzenetek:
+#-------------------------------------
+# What's the check in time?
+# What are the check-in and check-out times?
+# Are there any local restaurants you'd suggest?
+# What's the Wi-Fi password for the property?
+# What's the pin for the lockbox?
+# What is the check-out procedure?
+# What is the manager's name?     -->   NEM kell tudja! 
+

requirements.txt

@@ -0,0 +1,2 @@
+streamlit
+openai