Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
+import psycopg2
 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
@@ -85,9 +85,9 @@ def insert_feedback(conn, rating, comment):
             cur.execute('INSERT INTO feedback (timestamp, rating, comment) VALUES (%s, %s, %s)', 
                         (timestamp, int(rating), comment))
             conn.commit()
-            st.write("Feedback inserted successfully.")
+            st.success("Feedback inserted successfully.")
     except Exception as e:
-        st.write(f"Error inserting feedback: {e}")
+        st.error(f"Error inserting feedback: {e}")
 
 # Streamlit interface with improved aesthetics
 st.set_page_config(page_title="Alter-IA Chat", page_icon="🤖")
@@ -146,7 +146,6 @@ 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)
@@ -156,37 +155,43 @@ if submit_button:
         # Add rating and comment section
         st.markdown("---")
         st.markdown("#### Rate the Response:")
-        
-        # Custom star rating HTML
-        rating_html = """
-        <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"><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)
-        
-        # 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:")
-
-        # Submit feedback
-        feedback_button = st.button("Submit Feedback")
-
-        if feedback_button:
-            if comment.strip() == "":
-                st.warning("⚠ Please provide a comment.")
-            else:
-                st.success("Thank you for your feedback!")
-
-                # Store feedback in PostgreSQL
-                conn = create_connection()
-                insert_feedback(conn, rating, comment)
-                conn.close()
+
+        # Use Streamlit form for feedback
+        with st.form(key='feedback_form'):
+            # Custom star rating HTML
+            rating_html = """
+            <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="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>
+            <input type="hidden" id="selected-rating" name="rating" value="3">
+            <script>
+                const ratingLabels = document.querySelectorAll('.star-rating label');
+                ratingLabels.forEach(label => {
+                    label.addEventListener('click', () => {
+                        document.getElementById('selected-rating').value = label.previousElementSibling.value;
+                    });
+                });
+            </script>
+            """
+            st.markdown(rating_html, unsafe_allow_html=True)
+
+            rating = st.text_input("Selected Rating:", value="3", key="rating_input", label_visibility="hidden")
+            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
+                    conn = create_connection()
+                    insert_feedback(conn, rating, comment)
+                    conn.close()
 
     else:
         st.warning("⚠ Please enter a message.")