Update app.py

app.py

@@ -2,6 +2,8 @@ import gradio as gr
 import pandas as pd
 import gspread
 from google.auth import default
+import requests
+import time 
 
 # Authenticate and authorize with Google Sheets
 #creds, _ = default()
@@ -12,31 +14,65 @@ gc = gspread.service_account(filename='botresponse-6f1a8c749aa0.json')
 sheet_id = "18hnoTsEaGMWMael42MXubb-FzAe5jJB5RpaSolIXyb0"
 worksheet_name = "Sheet1"
 
-def feedback_response(feedback, comments, new_input_query):
+# 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'
-    else:
+    elif feedback == 'Nonsense':
         response = 'Nonsense'
-    
-    # Open the Google Sheets document
+    else:
+        response = 'NA'
+
+    # Update Google Sheets
     sh = gc.open_by_key(sheet_id)
     worksheet = sh.worksheet(worksheet_name)
-    
-    # Create a DataFrame from the response and additional comments
-    df = pd.DataFrame({'New Input Query': [new_input_query],'Response': [response], 'Additional Comments': [comments]})
-    
-    # Append the data to the worksheet
-    worksheet.append_rows(df.values.tolist())
-    
-    return "Your feedback has been recorded. Thank you!" 
+
+    # 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())
+
+    return initial_response, elapsed_time  # Return the initial_response and elapsed_time
 
 # Set up the Gradio Interface
 feedback_interface = gr.Interface(
     fn=feedback_response,
     inputs=[
+        gr.Textbox(label="Input Message"),
         gr.Radio(
             choices=[
                 "Informative",
@@ -45,13 +81,14 @@ feedback_interface = gr.Interface(
             ],
             label="Feedback"
         ),
-        gr.Textbox(label="New Input Query"),
         gr.Textbox(label="Additional Comments")
-        
     ],
-    outputs="text",
+    outputs=[
+        gr.Textbox(label="Output", type="text"),
+        gr.Textbox(label="Elapsed Time (s)", type="text")  # Change the type to "text" for two decimal places
+    ],
     title="Beta: Itell Guide Response Bot",
-    description = "This is an interface to test iTELL's guide on the side. Please be aware that responses can take up to 20 seconds"
+    description="This is an interface to test iTELL's guide on the side. Please be aware that responses can take up to 20 seconds",
 )
 
 # Launch the interface