Update app.py

app.py

@@ -67,7 +67,10 @@ qa = RetrievalQA.from_chain_type(
     chain_type_kwargs={"prompt": prompt},
 )
 
-# Streamlit UI setup
+mport streamlit as st
+from datetime import datetime
+
+# 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
@@ -75,48 +78,16 @@ def chatbot_response(user_input):
     response = qa.run(user_input)
     return response
 
-# Define function to save feedback to CSV
-def save_feedback(question, response, rating, comment):
-    try:
-        filename = 'feedback.csv'
-        file_exists = os.path.isfile(filename)
-
-        # Open CSV file and append the new feedback
-        with open(filename, 'a', newline='', encoding='utf-8') as csvfile:
-            fieldnames = ['question', 'response', 'rating', 'comment']
-            writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
-            if not file_exists:
-                writer.writeheader()
-            writer.writerow({'question': question, 'response': response, 'rating': rating, 'comment': comment})
-
-        st.success("Thank you for your feedback! It has been saved.")
-    except Exception as e:
-        st.error(f"Error saving feedback: {e}")
-        st.write(f"Exception: {e}")
-
-# Use session state to store user input, bot response, rating, and comment
-if 'user_input' not in st.session_state:
-    st.session_state.user_input = ""
-
-if 'bot_response' not in st.session_state:
-    st.session_state.bot_response = ""
-
-if 'rating' not in st.session_state:
-    st.session_state.rating = 3  # Default rating
-
-if 'comment' not in st.session_state:
-    st.session_state.comment = ""
-
 # Create columns for logos
 col1, col2, col3 = st.columns([2, 3, 2])
 
 with col1:
-    st.image("Design 3_22.png", width=150, use_column_width=True)
+    st.image("Design 3_22.png", width=150, use_column_width=True)  # Adjust image path and size as needed
 
 with col3:
-    st.image("Altereo logo 2023 original - eau et territoires durables.png", width=150, use_column_width=True)
+    st.image("Altereo logo 2023 original - eau et territoires durables.png", width=150, use_column_width=True)  # Adjust image path and size as needed
 
-# Add CSS for styling
+# CSS for styling
 st.markdown("""
     <style>
     .centered-text {
@@ -126,35 +97,80 @@ st.markdown("""
         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)
 
-# Center and color text
 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)
 
 # Input and button for user interaction
-st.session_state.user_input = st.text_input("You:", st.session_state.user_input)
-
-if st.button("Ask 📨"):
-    if st.session_state.user_input.strip() != "":
-        st.session_state.bot_response = chatbot_response(st.session_state.user_input)
-
-if st.session_state.bot_response:
-    st.markdown("### Bot:")
-    st.text_area("", value=st.session_state.bot_response, height=600)
-
-    # Feedback Section
-    st.markdown("### Évaluation de la réponse")
-    st.session_state.rating = st.slider("Rating (1 to 5)", 1, 5, st.session_state.rating)
-    st.session_state.comment = st.text_area("Your comment:", st.session_state.comment)
-    
-    if st.button("Submit Feedback"):
-        if st.session_state.comment.strip() != "":
-            save_feedback(st.session_state.user_input, st.session_state.bot_response, st.session_state.rating, st.session_state.comment)
-        else:
-            st.warning("⚠️ Please enter a comment.")
+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:")
+        
+        # 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>
+        """
+        st.markdown(rating_html, unsafe_allow_html=True)
+        
+        # Get the selected rating via JavaScript
+        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 (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")
+
+    else:
+        st.warning("⚠ Please enter a message.")
 
 # Motivational quote at the bottom
 st.markdown("---")
-st.markdown("La collaboration est la clé du succès. Chaque question trouve sa réponse, chaque défi devient une opportunité.")
+st.markdown("La collaboration est la clé du succès. Chaque question trouve sa réponse, chaque défi devient une opportunité.")