Update app.py

app.py

@@ -6,7 +6,7 @@ from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain_huggingface import HuggingFaceEndpoint
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain, RetrievalQA
-from huggingface_hub import login
+from huggingface_hub import login, Repository
 
 # Login to Hugging Face
 login(token=st.secrets["HF_TOKEN"])
@@ -74,14 +74,10 @@ st.set_page_config(page_title="Alter-IA Chat", page_icon="🤖")
 def chatbot_response(user_input):
     response = qa.run(user_input)
     return response
-import os
-import csv
-import streamlit as st
-from huggingface_hub import Repository, HfApi
 
 # Initialize the repository
-reposss_id = "mery22/testing"  # Replace with your repo ID
-reposss = Repository(local_dir=".", clone_from=reposss_id, token=st.secrets["HF_TOKEN"])
+repo_id = "mery22/testing"  # Replace with your repo ID
+repo = Repository(local_dir=".", clone_from=repo_id, token=st.secrets["HF_TOKEN"])
 
 # Define function to save feedback to CSV
 def save_feedback(question, response, rating, comment):
@@ -98,45 +94,13 @@ def save_feedback(question, response, rating, comment):
             writer.writerow({'question': question, 'response': response, 'rating': rating, 'comment': comment})
 
         # Push the updated file to the Hugging Face repository
-        reposss.push_to_hub(commit_message="Updated feedback")
+        repo.push_to_hub(commit_message="Updated feedback")
 
         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}")
 
-# Example usage:
-question = "What is the capital of France?"
-response = "The capital of France is Paris."
-rating = 5
-comment = "Good response!"
-
-# Save feedback and check the output in Streamlit
-save_feedback(question, response, rating, 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}")  # Debugging: Print any exceptions
-
-# Example usage:
-question = "What is the capital of France?"
-response = "The capital of France is Paris."
-rating = 5
-comment = "Good response!"
-
-# Save feedback and check the output in Streamlit
-save_feedback(question, response, rating, comment)
-
-# Optionally, display the file content to confirm it was written correctly
-if st.button("View Feedback File"):
-    try:
-        with open('/tmp/feedback.csv', 'r', encoding='utf-8') as csvfile:
-            st.text(csvfile.read())
-    except FileNotFoundError:
-        st.error("Feedback file not found.")
-
 # 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 = ""