Update Gradio_UI.py

Gradio_UI.py

@@ -26,7 +26,7 @@ from smolagents.utils import _is_package_available
 
 
 
-
+# Google Sheet connection
 import json
 import pygsheets
 import pandas as pd
@@ -56,7 +56,7 @@ df = pd.DataFrame()
 df['name'] = ['John', 'Steve', 'Sarah']
 
 #open the google spreadsheet (where 'PY to Gsheet Test' is the name of my sheet)
-sh = gc.open('PY to Gsheet Test')
+sh = gc.open('LLM_evaluator')
 
 #select the first sheet 
 wks = sh[0]
@@ -198,12 +198,25 @@ def stream_to_gradio(
         ):
             yield message
 
+    # Set up GoogleSheet
+    sheet_title = 'LLM_evaluator'  # Ensure this matches the sheet's name
+    sh = gc.open(sheet_title)
+    worksheet = sh.sheet1  # Use the first sheet
+    def append_to_google_sheet(query, answer):
+        """Append query and final answer to the Google Sheet."""
+        new_row = [query, answer]
+        worksheet.append_table(new_row, start="A1", end=None, dimension="ROWS")
+    
     final_answer = step_log  # Last log is the run's final_answer
     final_answer = handle_agent_output_types(final_answer)
 
     if isinstance(final_answer, AgentText):
         formatted_final_answer = f"**Query:** {task}\n\n**Final answer:**\n{final_answer.to_string()}\n"
 
+
+        # Append data to Google Sheet
+        append_to_google_sheet(task, final_answer.to_string())
+        
         yield gr.ChatMessage(
             role="assistant",
             content=formatted_final_answer,