Update app.py

app.py

@@ -78,14 +78,16 @@ def create_connection():
         port=os.getenv("DB_PORT")
     )
 
-def insert_feedback(conn, rating, comment):
+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
+        conn.close()
+        return True
     except Exception as e:
         st.error(f"Error inserting feedback: {e}")
         return False
@@ -159,28 +161,7 @@ if submit_button:
 
         # 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")
+            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")
@@ -190,10 +171,10 @@ if submit_button:
                     st.warning("⚠ Please provide a comment.")
                 else:
                     # Store feedback in PostgreSQL
-                    conn = create_connection()
-                    if insert_feedback(conn, rating, comment):
+                    if insert_feedback(rating, comment):
                         st.success("Thank you for your feedback!")
-                    conn.close()
+                    else:
+                        st.error("Failed to submit feedback.")
 
     else:
         st.warning("⚠ Please enter a message.")