Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st 
 from langchain_community.llms import HuggingFaceTextGenInference
 import os
+import io
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 from langchain.schema import StrOutputParser
 # from datetime import datetime
@@ -21,7 +22,7 @@ os.environ['LANGCHAIN_TRACING_V2'] = "true"
 
 
 API_TOKEN = os.getenv('HF_INFER_API')
-POSTGRE_URL = os.environ['POSTGRE_URL']
+# POSTGRE_URL = os.environ['POSTGRE_URL']
 
 @st.cache_resource
 def get_llm_chain():
@@ -33,29 +34,29 @@ def get_llm_chain():
     )
 
 
-@st.cache_resource
-def get_db_connection(conn_url, password=None):
+# @st.cache_resource
+# def get_db_connection(conn_url, password=None):
     
-    url = up.urlparse(conn_url)
-
-    conn = psycopg2.connect(
-        database=url.path[1:],
-        user=url.username,
-        password=password if password is not None else url.password,
-        host=url.hostname,
-        port=url.port
-    )
+#     url = up.urlparse(conn_url)
+
+#     conn = psycopg2.connect(
+#         database=url.path[1:],
+#         user=url.username,
+#         password=password if password is not None else url.password,
+#         host=url.hostname,
+#         port=url.port
+#     )
 
-    print("Connection to database succesfull!")
-    return conn
+#     print("Connection to database succesfull!")
+#     return conn
 
 # @st.cache_resource
 # def get_memory():
 #     return PostgresChatMessageHistory(connection_string=POSTGRE_URL, session_id=str(datetime.timestamp(datetime.now())))
 
 
-if 'conn' not in st.session_state:
-    st.session_state.conn = get_db_connection(POSTGRE_URL)
+# if 'conn' not in st.session_state:
+#     st.session_state.conn = get_db_connection(POSTGRE_URL)
 
 # if 'cursor' not in st.session_state:
 #     st.session_state.cursor = st.session_state.conn.cursor()
@@ -113,25 +114,54 @@ if prompt := st.chat_input("Ask me anything.."):
     # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})
 
-    # Insert data into the table
-    try :
-        try :
-            cur = st.session_state.conn.cursor()
-        except:
-            get_db_connection.clear()
-            st.session_state.conn = get_db_connection(POSTGRE_URL)
-            cur = st.session_state.conn.cursor()
-            
-        cur.execute(
-            f"INSERT INTO chat_history (input_text, response_text, created_at) VALUES (%s, %s, %s)",
-            (prompt, response, datetime.now(timezone.utc) + timedelta(hours=7))
-        )
+    try:
+        timestamp = datetime.now(timezone.utc) + timedelta(hours=7)
+        chat_text = f"Timestamp: {timestamp}\nUser Input: {prompt}\nChatbot Response: {response}\n\n"
+        text_file = io.StringIO(chat_text)  # Use io.StringIO
+
+        data = {
+            "text_content": [chat_text]  # Store the raw text
+        }
+        dataset = Dataset.from_dict(data)
+
+        # dataset_name = "your_dataset_name"  # Replace with your desired dataset name
+        # dataset_name = os.environ["DB_NAME"]
+        dataset_name = "chat_with_me_history"
+        repo_id = f"jonathanjordan21/{dataset_name}"  # Full repo ID
         
-        # Commit the transaction
-        st.session_state.conn.commit()
-        cur.close()
+        dataset.push_to_hub(
+            repo_id=repo_id,
+            private=True,  # Set to False if you want it to be public
+            # token="your_huggingface_token",  # Replace with your token
+            token=API_TOKEN
+        )
+        print(f"Chat history added to Hugging Face dataset: {repo_id}")
+
+
     except Exception as e:
         print("ERROR!!!\n", str(e))
         print("User Input :", prompt)
         print("Chatbot Response :", response)
+
+    # # Insert data into the table
+    # try :
+    #     try :
+    #         cur = st.session_state.conn.cursor()
+    #     except:
+    #         get_db_connection.clear()
+    #         st.session_state.conn = get_db_connection(POSTGRE_URL)
+    #         cur = st.session_state.conn.cursor()
+            
+    #     cur.execute(
+    #         f"INSERT INTO chat_history (input_text, response_text, created_at) VALUES (%s, %s, %s)",
+    #         (prompt, response, datetime.now(timezone.utc) + timedelta(hours=7))
+    #     )
+        
+    #     # Commit the transaction
+    #     st.session_state.conn.commit()
+    #     cur.close()
+    # except Exception as e:
+    #     print("ERROR!!!\n", str(e))
+    #     print("User Input :", prompt)
+    #     print("Chatbot Response :", response)