Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
-import csv
 import streamlit as st
+from datetime import datetime
+import psycopg2
 from langchain_community.vectorstores import FAISS
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_huggingface import HuggingFaceEndpoint
@@ -67,38 +68,30 @@ 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(
+    return 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()
+
+def insert_feedback(conn, rating, comment):
+    try:
+        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()
+            st.write("Feedback inserted successfully.")
+    except Exception as e:
+        st.write(f"Error inserting feedback: {e}")
 
 # Streamlit interface with improved aesthetics
 st.set_page_config(page_title="Alter-IA Chat", page_icon="🤖")
 
-# Define function to handle user input and display chatbot response
 def chatbot_response(user_input):
     response = qa.run(user_input)
     return response
@@ -168,14 +161,14 @@ if submit_button:
         <div class="star-rating">
             <input type="radio" id="5-stars" name="rating" value="5"><label for="5-stars">★</label>
             <input type="radio" id="4-stars" name="rating" value="4"><label for="4-stars">★</label>
-            <input type="radio" id="3-stars" name="rating" value="3" checked><label for="3-stars">★</label>
+            <input type="radio" id="3-stars" name="rating" value="3"><label for="3-stars">★</label>
             <input type="radio" id="2-stars" name="rating" value="2"><label for="2-stars">★</label>
             <input type="radio" id="1-star" name="rating" value="1"><label for="1-star">★</label>
         </div>
         """
         st.markdown(rating_html, unsafe_allow_html=True)
         
-        # Get the selected rating via JavaScript
+        # Use a hidden input to capture the selected rating
         rating = st.text_input("Selected Rating:", value="3", key="rating_input", label_visibility="hidden")
 
         comment = st.text_area("Your Comment:")
@@ -188,14 +181,10 @@ if submit_button:
                 st.warning("⚠ Please provide a comment.")
             else:
                 st.success("Thank you for your feedback!")
-                
+
                 # 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()
+                insert_feedback(conn, rating, comment)
                 conn.close()
 
     else: