add feedback and sheets upload

.gitignore

@@ -1 +1,3 @@
-.env
+.env
+# Service account credentials
+credentials.json

requirements.txt

@@ -11,3 +11,6 @@ sentence-transformers>=2.7.0
 # LLM providers
 openai
 together>=0.2.8
+
+gspread
+oauth2client

src/app.py

@@ -5,6 +5,9 @@ from together import Together
 import time
 import os
 from dotenv import load_dotenv
+from google_sheets_uploader import upload_to_google_sheets
+import pandas as pd
+import json
 
 load_dotenv()
 
@@ -14,6 +17,10 @@ if "current_embedder_name" not in st.session_state:
     st.session_state.current_embedder_name = None
 if "last_sources" not in st.session_state:
     st.session_state.last_sources = []
+if "session_data" not in st.session_state:
+    st.session_state.session_data = []
+if "uploaded_rows_count" not in st.session_state:
+    st.session_state.uploaded_rows_count = 0
 
 st.set_page_config(
     page_title="Bipolar Assistant Chatbot",
@@ -70,6 +77,18 @@ with st.sidebar:
     top_p = st.slider('top_p', min_value=0.01, max_value=1.0, value=0.9, step=0.01)
     max_length = st.slider('max_length', min_value=100, max_value=1000, value=500, step=10)
 
+    if st.button("Save and Upload to Google Sheets"):
+        try:
+            new_data = st.session_state.session_data[st.session_state.uploaded_rows_count:]
+            if new_data:
+                upload_to_google_sheets(new_data)
+                st.session_state.uploaded_rows_count = len(st.session_state.session_data)
+                st.success("Successfully uploaded to Google Sheets!")
+            else:
+                st.info("No new data to upload.")
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
+
 st.markdown("## 💬 Bipolar Assistant Chatbot")
 
 if "messages" not in st.session_state:
@@ -95,9 +114,28 @@ with sources_col:
             st.markdown("*Sources will appear here after you ask a question.*")
 
 with chat_col:
-    for message in st.session_state.messages:
+    for i, message in enumerate(st.session_state.messages):
         with st.chat_message(message["role"]):
             st.markdown(message["content"])
+            if message["role"] == "assistant" and i > 0:
+                feedback_options = ["Good", "Bad", "Neutral"]
+                answer_feedback = st.radio(
+                    "Rate your answer:",
+                    options=feedback_options,
+                    index=2,  # Default to Neutral
+                    key=f"answer_feedback_{i}",
+                    horizontal=True,
+                )
+                st.session_state.session_data[i // 2 - 1]["feedback"] = answer_feedback
+
+                source_feedback = st.radio(
+                    "Rate your sources:",
+                    options=feedback_options,
+                    index=2,  # Default to Neutral
+                    key=f"source_feedback_{i}",
+                    horizontal=True,
+                )
+                st.session_state.session_data[i // 2 - 1]["source_feedback"] = source_feedback
 
     if user_input := st.chat_input("Ask me questions about the CANMAT bipolar guideline!"):
         st.chat_message("user").markdown(user_input)
@@ -134,4 +172,14 @@ with chat_col:
 
         st.session_state.messages.append({"role": "assistant", "content": collected})
 
+        st.session_state.session_data.append(
+            {
+                "query": user_input,
+                "response": collected,
+                "sources": json.dumps(st.session_state.last_sources, indent=4),
+                "feedback": "Neutral",
+                "source_feedback": "Neutral",
+            }
+        )
+
         st.rerun()

src/google_sheets_uploader.py

@@ -0,0 +1,47 @@
+
+import gspread
+from oauth2client.service_account import ServiceAccountCredentials
+import pandas as pd
+import os
+import json
+
+def get_gspread_client():
+    scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
+    
+    google_creds_json = os.environ.get("GOOGLE_CREDENTIALS_JSON")
+
+    if google_creds_json:
+        # Use credentials from Hugging Face secret
+        creds_dict = json.loads(google_creds_json)
+        creds = ServiceAccountCredentials.from_json_keyfile_dict(creds_dict, scope)
+    else:
+        # Fallback to local credentials file
+        creds = ServiceAccountCredentials.from_json_keyfile_name("credentials.json", scope)
+        
+    client = gspread.authorize(creds)
+    return client
+
+def upload_to_google_sheets(session_data):
+    client = get_gspread_client()
+    sheet = client.open("Bipolar Chatbot Logs").sheet1
+    
+    # Convert session_data to a pandas DataFrame
+    df = pd.DataFrame(session_data)
+    
+    # Get existing headers from the sheet
+    try:
+        existing_headers = sheet.row_values(1)
+    except gspread.exceptions.APIError:
+        existing_headers = []
+
+    # Define column order
+    columns = ['query', 'response', 'sources', 'feedback', 'source_feedback']
+    df = df[columns]
+
+    # If the sheet is empty, write the headers
+    if not existing_headers:
+        sheet.append_row(columns)
+    
+    # Append the DataFrame to the sheet
+    for row in df.itertuples(index=False, name=None):
+        sheet.append_row(list(row))