Update app.py

app.py

@@ -6,7 +6,82 @@ import requests
 import time 
 import json
 
-# ... (rest of your code)
+# Authenticate and authorize with Google Sheets
+# creds, _ = default()
+# gc = gspread.authorize(creds)
+gc = gspread.service_account(filename='botresponse-6f1a8c749aa0.json')
+    
+# Specify your Google Sheets credentials, sheet_id, and worksheet_name
+sheet_id = "18hnoTsEaGMWMael42MXubb-FzAe5jJB5RpaSolIXyb0"
+worksheet_name = "Sheet1"
+
+# Function to get the initial response
+def get_initial_response(input_message):
+    url = "https://itell-api.learlab.vanderbilt.edu/chat"
+
+    payload = {
+        "textbook_name": "think-python-2e",
+        "message": input_message
+    }
+    headers = {"Content-Type": "application/json"}
+
+    # Measure the start time
+    start_time = time.time()
+
+    response = requests.post(url, json=payload, headers=headers)
+    data = json.loads(response.text)
+    message = data['message']
+
+    # Calculate the elapsed time
+    elapsed_time = time.time() - start_time
+    elapsed_time = round(elapsed_time, 2)
+    response_time_message = f"Response time: {elapsed_time} seconds"
+
+    return message, response_time_message  # Return the initial_response and elapsed_time
+
+# Function to collect feedback and update the spreadsheet
+def feedback_response(input_message, feedback, add_feedback, comments):
+    # Get the initial response and elapsed_time
+    initial_response, elapsed_time = get_initial_response(input_message)
+
+    # Collect feedback
+    response = ''
+    if feedback == 'Good':
+        response = 'Good'
+    elif feedback == 'Bad':
+        response = 'Bad'
+    elif feedback == 'Inappropriate':
+        response = 'Inappropriate'
+    else:
+        response = 'NA'
+    
+    # More feedback
+    # Collect feedback
+    add_response = ''
+    if add_feedback == 'Informative':
+        add_response = 'Informative'
+    elif add_feedback == 'Inaccurate':
+        add_response = 'Inaccurate'
+    elif add_feedback == 'Nonsense':
+        add_response = 'Nonsense'
+    else:
+        add_response = 'NA'
+
+    # Update Google Sheets
+    sh = gc.open_by_key(sheet_id)
+    worksheet = sh.worksheet(worksheet_name)
+    
+    thankyou = "Thank you for your feedback. Your response and feedback has been recorded!"
+    # Append the data to the worksheet only if comments have a value
+    if comments:
+        # Create a DataFrame from the response and additional comments
+        df = pd.DataFrame({'Input Message': [input_message], 'Output': [initial_response], 'Time took in Seconds': [elapsed_time],'Feedback': [response], 'add_feedback': [add_response], 'Additional Comments': [comments]})
+
+        # Append the data to the worksheet
+        worksheet.append_rows(df.values.tolist())
+        initial_response = thankyou
+    
+    return initial_response, elapsed_time  # Return the initial_response and elapsed_time
 
 # Set up the Gradio Interface
 feedback_interface = gr.Interface(