Update app.py

app.py

@@ -3,18 +3,83 @@ import pandas as pd
 import gspread
 from google.auth import default
 import requests
-import time 
+import time
 import json
 
-# ... (The rest of your code remains the same)
+# Authenticate and authorize with Google Sheets
+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, comments):
+    # Get the initial response and elapsed_time
+    initial_response, elapsed_time = get_initial_response(input_message)
+
+    # Collect feedback
+    response = ''
+    if feedback == 'Informative':
+        response = 'Informative'
+    elif feedback == 'Inaccurate':
+        response = 'Inaccurate'
+    elif feedback == 'Nonsense':
+        response = 'Nonsense'
+    elif feedback == 'Good':
+        response = 'Good'
+    elif feedback == 'Bad':
+        response = 'Bad'
+    elif feedback == 'Inappropriate':
+        response = 'Inappropriate'
+    else:
+        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 have 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],
+                           '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
 
 # Define custom CSS styles for the buttons
 custom_css = """
 <style>
-    .button-container {
-        display: flex;
-        justify-content: space-between;
-    }
     .feedback-button {
         background-color: #008CBA;
         color: white;
@@ -34,28 +99,33 @@ custom_css = """
 </style>
 """
 
-# Set up the Gradio Interface
+# Define the custom HTML for the buttons
+custom_html = custom_css + """
+<div>
+    <label>Feedback (Top Row):</label>
+    <button class="feedback-button" onclick="setFeedback('Good')">Good</button>
+    <button class="feedback-button" onclick="setFeedback('Bad')">Bad</button>
+    <button class="feedback-button" onclick="setFeedback('Inappropriate')">Inappropriate</button>
+</div>
+<div>
+    <label>Feedback (Bottom Row):</label>
+    <button class="feedback-button" onclick="setFeedback('Informative')">Informative</button>
+    <button class="feedback-button" onclick="setFeedback('Inaccurate')">Inaccurate</button>
+    <button class="feedback-button" onclick="setFeedback('Nonsense')">Nonsense</button>
+</div>
+"""
+
+# Set up the Gradio Interface with HTML components for custom buttons
 feedback_interface = gr.Interface(
     fn=feedback_response,
     inputs=[
         gr.Textbox(label="Input Message"),
+        gr.HTML(label="Feedback Buttons", html=custom_html),
         gr.Textbox(label="Additional Comments")
     ],
     outputs=[
-        gr.HTML(label="Feedback (Top Row)", html=custom_css + """
-            <div class="button-container">
-                <button class="feedback-button" id="good-feedback" onclick="setFeedback('Good')">Good</button>
-                <button class="feedback-button" id="bad-feedback" onclick="setFeedback('Bad')">Bad</button>
-                <button class="feedback-button" id="inappropriate-feedback" onclick="setFeedback('Inappropriate')">Inappropriate</button>
-            </div>
-        """),
-        gr.HTML(label="Feedback (Bottom Row)", html=custom_css + """
-            <div class="button-container">
-                <button class="feedback-button" id="informative-feedback" onclick="setFeedback('Informative')">Informative</button>
-                <button class="feedback-button" id="inaccurate-feedback" onclick="setFeedback('Inaccurate')">Inaccurate</button>
-                <button class="feedback-button" id="nonsense-feedback" onclick="setFeedback('Nonsense')">Nonsense</button>
-            </div>
-        """)
+        gr.Textbox(label="Output", type="text"),
+        gr.Textbox(label="Elapsed Time (s)", type="text")
     ],
     title="Beta: Itell Guide Response Bot",
     description="""