Update app.py

app.py

@@ -68,34 +68,7 @@ qa = RetrievalQA.from_chain_type(
     chain_type_kwargs={"prompt": prompt},
 )
 
-# PostgreSQL connection setup using secrets from Hugging Face Spaces
-def create_connection():
-    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")
-    )
 
-def insert_feedback(rating, comment):
-    try:
-        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()
-        return True
-    except Exception as e:
-        st.error(f"Error inserting feedback: {e}")
-        return False
-    finally:
-        if conn:
-            conn.close()
-
-# Streamlit interface with improved aesthetics
-st.set_page_config(page_title="Alter-IA Chat", page_icon="🤖")
 
 def chatbot_response(user_input):
     response = qa.run(user_input)
@@ -110,39 +83,7 @@ with col1:
 with col3:
     st.image("Altereo logo 2023 original - eau et territoires durables.png", width=150, use_column_width=True)
 
-# CSS for styling
-st.markdown("""
-    <style>
-    .centered-text {
-        text-align: center;
-    }
-    .centered-orange-text {
-        text-align: center;
-        color: darkorange;
-    }
-    .star-rating {
-        display: flex;
-        flex-direction: row-reverse;
-        justify-content: center;
-        cursor: pointer;
-    }
-    .star-rating input[type="radio"] {
-        display: none;
-    }
-    .star-rating label {
-        font-size: 2em;
-        color: #ddd;
-        padding: 0 5px;
-        transition: color 0.3s;
-    }
-    .star-rating input[type="radio"]:checked ~ label {
-        color: gold;
-    }
-    .star-rating input[type="radio"]:hover ~ label {
-        color: gold;
-    }
-    </style>
-    """, unsafe_allow_html=True)
+
 
 st.markdown('<h3 class="centered-text">🤖 AlteriaChat 🤖 </h3>', unsafe_allow_html=True)
 st.markdown('<p class="centered-orange-text">"Votre Réponse à Chaque Défi Méthodologique "</p>', unsafe_allow_html=True)
@@ -151,34 +92,81 @@ st.markdown('<p class="centered-orange-text">"Votre Réponse à Chaque Défi Mé
 user_input = st.text_input("You:", "")
 submit_button = st.button("Ask 📨")
 
-if submit_button:
-    if user_input.strip() != "":
-        bot_response = chatbot_response(user_input)
-        st.markdown("### Bot:")
-        st.text_area("", value=bot_response, height=300)
-
-        # Add rating and comment section
-        st.markdown("---")
-        st.markdown("#### Rate the Response:")
-
-        # Use Streamlit form for feedback
-        with st.form(key='feedback_form'):
-            rating = st.select_slider("Rating:", options=[1, 2, 3, 4, 5], value=3)
-            comment = st.text_area("Your Comment:")
-
-            submit_feedback_button = st.form_submit_button("Submit Feedback")
-
-            if submit_feedback_button:
-                if comment.strip() == "":
-                    st.warning("⚠ Please provide a comment.")
-                else:
-                    # Store feedback in PostgreSQL
-                    if insert_feedback(rating, comment):
-                        st.success("Thank you for your feedback!")
-                    else:
-                        st.error("Failed to submit feedback.")
+import os
+import streamlit as st
+import psycopg2
+from datetime import datetime
+
+# Function to create a connection to PostgreSQL
+def create_connection():
+    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")
+    )
+
+# Function to create the feedback table if it doesn't exist
+def create_feedback_table(conn):
+    cursor = conn.cursor()
+    cursor.execute("""
+        CREATE TABLE IF NOT EXISTS feedback (
+            id SERIAL PRIMARY KEY,
+            user_input TEXT NOT NULL,
+            bot_response TEXT NOT NULL,
+            rating INT CHECK (rating >= 1 AND rating <= 5),
+            comment TEXT,
+            timestamp TIMESTAMP DEFAULT CURRENT_TIMESTAMP
+        );
+    """)
+    conn.commit()
+    cursor.close()
+
+# Function to insert feedback into the database
+def insert_feedback(conn, user_input, bot_response, rating, comment):
+    cursor = conn.cursor()
+    cursor.execute(
+        "INSERT INTO feedback (user_input, bot_response, rating, comment, timestamp) VALUES (%s, %s, %s, %s, %s)",
+        (user_input, bot_response, rating, comment, datetime.now())
+    )
+    conn.commit()
+    cursor.close()
+
+# Initialize connection and create the table if necessary
+conn = create_connection()
+create_feedback_table(conn)
+
+# Streamlit app UI and logic
+st.markdown("## Rate your experience")
+
+# Create a star-based rating system using radio buttons
+rating = st.radio(
+    "Rating",
+    options=[1, 2, 3, 4, 5],
+    format_func=lambda x: "★" * x  # Display stars based on the rating
+)
+
+# Text area for leaving a comment
+comment = st.text_area("Leave a comment")
+
+# Display bot response and user input for context
+st.markdown("### Your Question:")
+st.write(user_input)
+st.markdown("### Bot's Response:")
+st.write(bot_response)
+
+# Submit feedback
+if st.button("Submit Feedback"):
+    if rating and comment:
+        insert_feedback(conn, user_input, bot_response, rating, comment)
+        st.success("Thank you for your feedback!")
     else:
-        st.warning("⚠ Please enter a message.")
+        st.warning("Please provide a rating and a comment.")
+
+# Close the connection when done
+conn.close()
+
 
 # Motivational quote at the bottom
 st.markdown("---")