Upload psychochat.py

psychochat.py

@@ -0,0 +1,141 @@
+import streamlit as st
+import os
+import pickle
+import time
+import tempfile
+import os
+
+import requests
+
+
+
+st.markdown(
+    """
+    <style>
+        .title {
+            text-align: center;
+            font-size: 2em;
+            font-weight: bold;
+        }
+    </style>
+    <div class="title">🧠 Phsycadalleics Chatbot</div>
+    """, 
+    unsafe_allow_html=True
+)
+if 'chat_history' not in st.session_state:
+    st.session_state.chat_history = []  # Define chat_history in the session state
+
+# Load and Save Conversations
+conversations_file = "conversations.pkl"
+
+if not os.path.exists(conversations_file):
+    # Create the file
+    with open(conversations_file, 'w') as f:
+        pass  # The file is created with this line, no need to write anything to it
+
+
+@st.cache_data
+def load_conversations():
+    try:
+        with open(conversations_file, "rb") as f:
+            return pickle.load(f)
+    except (FileNotFoundError, EOFError):
+        return []
+
+def save_conversations(conversations):
+    temp_conversations_file = "temp_" + conversations_file
+    with open(temp_conversations_file, "wb") as f:
+        pickle.dump(conversations, f)
+    os.replace(temp_conversations_file, conversations_file)
+
+
+if 'conversations' not in st.session_state:
+    st.session_state.conversations = load_conversations()
+
+if 'current_conversation' not in st.session_state:
+    st.session_state.current_conversation = [{"role": "assistant", "content": "How may I assist you today?"}]
+
+
+def truncate_string(s, length=30):
+    return s[:length].rstrip() + "..." if len(s) > length else s
+
+
+def display_chats_sidebar():
+    with st.sidebar.container():
+        st.header('Settings')
+        col1, col2 = st.columns([1, 1])
+
+        with col1:
+            if col1.button('Start New Chat', key="new_chat"):
+                st.session_state.current_conversation = []
+                st.session_state.conversations.append(st.session_state.current_conversation)
+                st.session_state.chat_history = []
+
+        with col2:
+            if col2.button('Clear All Chats', key="clear_all"):
+                st.session_state.conversations = []
+                st.session_state.current_conversation = []
+
+    with st.sidebar.container():
+        st.header('Conversations')
+        for idx, conversation in enumerate(st.session_state.conversations):
+            if conversation:
+                chat_title_raw = next((msg["content"] for msg in conversation if msg["role"] == "user"), "New Chat")
+                chat_title = truncate_string(chat_title_raw)
+                if st.sidebar.button(f"{chat_title}", key=f"chat_button_{idx}"):
+                    st.session_state.current_conversation = st.session_state.conversations[idx]
+                    
+
+def dummy_response(prompt_input, chat_history):
+    api_url = 'http://b5c9-34-125-138-225.ngrok.io/api/chat'
+
+    data_payload = {
+        'query': prompt_input,
+        'chat_history': chat_history 
+    }
+
+    response = requests.post(api_url, json=data_payload)
+
+    if response.status_code == 200:
+        response_data = response.json()
+        result = response_data.get('response')
+        st.session_state.chat_history = response_data.get('chat_history')
+    else:
+        # Handle the case where the API does not return a successful response
+        result = f"An error occurred: {response.text}"
+    
+    return result, chat_history
+
+    
+
+def main_app():
+    for message in st.session_state.current_conversation:
+        with st.chat_message(message["role"]):
+            st.write(message["content"])
+
+    if prompt := st.chat_input('Send a Message'):
+        st.session_state.current_conversation.append({"role": "user", "content": prompt})
+        print(' PROMPT :', prompt)
+        with st.chat_message("user"):
+            st.write(prompt)
+
+        with st.chat_message("assistant"):
+            with st.spinner("Thinking..."):
+                
+
+                response, st.session_state.chat_history = dummy_response(prompt, st.session_state.chat_history) 
+                full_response = ''
+                placeholder = st.empty()
+                for item in response:
+                    full_response += item
+                    time.sleep(0.003)
+                    placeholder.markdown(full_response)
+                placeholder.markdown(full_response)
+                print(' RESPONSE :', full_response)
+
+                st.session_state.current_conversation.append({"role": "assistant", "content": response})
+                save_conversations(st.session_state.conversations)
+        
+        
+display_chats_sidebar()
+main_app()