Update app.py

app.py

@@ -67,7 +67,33 @@ qa = RetrievalQA.from_chain_type(
     chain_type_kwargs={"prompt": prompt},
 )
 
+import psycopg2
+import streamlit as st
 from datetime import datetime
+import os
+
+# PostgreSQL connection setup using secrets from Hugging Face Spaces
+def create_connection():
+    conn = psycopg2.connect(
+        host=os.getenv("DB_HOST"),
+        database=os.getenv("DB_NAME"),
+        user=os.getenv("DB_USER"),
+        password=os.getenv("DB_PASSWORD"),
+        port=os.getenv("DB_PORT")
+    )
+    return conn
+
+def create_table(conn):
+    with conn.cursor() as cur:
+        cur.execute('''
+        CREATE TABLE IF NOT EXISTS feedback (
+            id SERIAL PRIMARY KEY,
+            timestamp TIMESTAMP NOT NULL,
+            rating INTEGER NOT NULL,
+            comment TEXT NOT NULL
+        );
+        ''')
+        conn.commit()
 
 # Streamlit interface with improved aesthetics
 st.set_page_config(page_title="Alter-IA Chat", page_icon="🤖")
@@ -81,10 +107,10 @@ def chatbot_response(user_input):
 col1, col2, col3 = st.columns([2, 3, 2])
 
 with col1:
-    st.image("Design 3_22.png", width=150, use_column_width=True)  # Adjust image path and size as needed
+    st.image("Design 3_22.png", width=150, use_column_width=True)
 
 with col3:
-    st.image("Altereo logo 2023 original - eau et territoires durables.png", width=150, use_column_width=True)  # Adjust image path and size as needed
+    st.image("Altereo logo 2023 original - eau et territoires durables.png", width=150, use_column_width=True)
 
 # CSS for styling
 st.markdown("""
@@ -163,9 +189,14 @@ if submit_button:
             else:
                 st.success("Thank you for your feedback!")
                 
-                # Store feedback (you can replace this part with code to save to a database or file)
-                with open("feedback.txt", "a") as f:
-                    f.write(f"{datetime.now()} | Rating: {rating} | Comment: {comment}\n")
+                # Store feedback in PostgreSQL
+                conn = create_connection()
+                with conn.cursor() as cur:
+                    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+                    cur.execute('INSERT INTO feedback (timestamp, rating, comment) VALUES (%s, %s, %s)', 
+                                (timestamp, int(rating), comment))
+                    conn.commit()
+                conn.close()
 
     else:
         st.warning("⚠ Please enter a message.")