Add gsheet logging

backend/german_polish.py

@@ -1,5 +1,6 @@
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
+from backend.logger.logger import log_to_gsheet
 
 model_id = "DebasishDhal99/polish-to-german-toponym-model-opus-mt-pl-de"
 
@@ -17,4 +18,5 @@ def polish_to_german_translation(polish_place):
         outputs = model.generate(**inputs, max_length=50)
     
     german_place = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    log_to_gsheet(polish_place, german_place, "Polish", "German")
     return german_place

backend/logger/logger.py

@@ -0,0 +1,21 @@
+import json
+import os
+from datetime import datetime
+import gspread
+from oauth2client.service_account import ServiceAccountCredentials
+
+scope = [
+    "https://spreadsheets.google.com/feeds",
+    "https://www.googleapis.com/auth/drive"
+]
+
+creds_dict = json.loads(os.environ["GOOGLE_SHEETS_CREDS_JSON"])
+creds = ServiceAccountCredentials.from_json_keyfile_dict(creds_dict, scope)
+client = gspread.authorize(creds)
+
+sheet = client.open("Topnym-Translation-Logger").worksheet("Sheet1")
+
+def log_to_gsheet(user_input: str, model_output: str, input_language: str, output_language: str):
+    timestamp = datetime.now().isoformat()
+    row = [timestamp, user_input, model_output, input_language, output_language]
+    sheet.append_row(row)

requirements.txt

@@ -1,4 +1,6 @@
 gradio==5.6.0
 torch==2.5.1
 transformers==4.51.1
-sentencepiece==0.2.0
+sentencepiece==0.2.0
+gspread==6.2.0
+ouath2client==4.1.3