Add Google Sheets storage for persistent feedback

- Add sheets_storage.py module for Google Sheets integration
- Modify demo.py to save ratings and notes to Google Sheets
- Add gspread and google-auth dependencies

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

ir/demo.py

@@ -19,6 +19,18 @@ import base64
 
 import gradio as gr
 
+# Google Sheets storage for persistent feedback
+try:
+    from . import sheets_storage
+    SHEETS_ENABLED = sheets_storage.is_sheets_enabled()
+except ImportError:
+    try:
+        import sheets_storage
+        SHEETS_ENABLED = sheets_storage.is_sheets_enabled()
+    except ImportError:
+        SHEETS_ENABLED = False
+        sheets_storage = None
+
 # ============================================================
 # UAE Theme Assets
 # ============================================================
@@ -854,7 +866,28 @@ def save_entity_rating(
         "page": current_page,
     }
 
-    # Save to file - use /data for HF Spaces persistence
+    emoji = "👍" if "relevant" in rating or "helpful" in rating else "👎"
+
+    # Try Google Sheets first (persistent)
+    if SHEETS_ENABLED and sheets_storage:
+        try:
+            success = sheets_storage.save_rating_to_sheets(
+                query=query,
+                category=category or "Not selected",
+                entity_id=entity_id,
+                entity_name=entity_name,
+                rank=rank,
+                score=score,
+                rating=rating,
+                page=current_page,
+                client_ip=client_ip
+            )
+            if success:
+                return f"{emoji} Rated #{rank} {entity_name[:20]}... as {rating} (saved to cloud)"
+        except Exception as e:
+            print(f"Google Sheets save failed: {e}")
+
+    # Fallback to local file
     if Path("/data").exists():
         feedback_file = Path("/data/demo_feedback.json")
     else:
@@ -872,8 +905,8 @@ def save_entity_rating(
         with open(feedback_file, "w", encoding="utf-8") as f:
             json.dump(all_feedback, f, ensure_ascii=False, indent=2)
 
-        emoji = "👍" if "relevant" in rating or "helpful" in rating else "👎"
-        return f"{emoji} Rated #{rank} {entity_name[:20]}... as {rating}"
+        storage_note = " (local only)" if not SHEETS_ENABLED else ""
+        return f"{emoji} Rated #{rank} {entity_name[:20]}... as {rating}{storage_note}"
 
     except Exception as e:
         return f"❌ Save failed: {str(e)}"
@@ -904,6 +937,21 @@ def save_notes_feedback(
         "notes": notes,
     }
 
+    # Try Google Sheets first (persistent)
+    if SHEETS_ENABLED and sheets_storage:
+        try:
+            success = sheets_storage.save_notes_to_sheets(
+                query=query,
+                category=category or "Not selected",
+                notes=notes,
+                client_ip=client_ip
+            )
+            if success:
+                return f"✅ Notes saved to cloud!"
+        except Exception as e:
+            print(f"Google Sheets save failed: {e}")
+
+    # Fallback to local file
     if Path("/data").exists():
         feedback_file = Path("/data/demo_feedback.json")
     else:
@@ -921,7 +969,8 @@ def save_notes_feedback(
         with open(feedback_file, "w", encoding="utf-8") as f:
             json.dump(all_feedback, f, ensure_ascii=False, indent=2)
 
-        return f"✅ Notes saved! Total: {len(all_feedback)} entries"
+        storage_note = " (local only)" if not SHEETS_ENABLED else ""
+        return f"✅ Notes saved!{storage_note} Total: {len(all_feedback)} entries"
 
     except Exception as e:
         return f"❌ Save failed: {str(e)}"

ir/sheets_storage.py

@@ -0,0 +1,196 @@
+"""
+Google Sheets storage for UAE KB feedback data.
+
+This module provides persistent storage for demo feedback using Google Sheets,
+which allows easy viewing and analysis of user feedback data.
+"""
+
+import os
+import json
+from datetime import datetime
+from typing import Optional, Dict, Any, List
+import logging
+
+logger = logging.getLogger(__name__)
+
+# Google Sheets configuration
+SHEET_ID = "1wwnpQRx32nbmApNK1LyvzUDIVx-tW4mpwgRPLvjm4gw"
+RATINGS_WORKSHEET = "ratings"
+NOTES_WORKSHEET = "notes"
+
+# Lazy-loaded client
+_sheets_client = None
+_spreadsheet = None
+
+
+def _get_credentials():
+    """Get Google credentials from environment variable."""
+    creds_json = os.environ.get("GOOGLE_SHEETS_CREDENTIALS")
+    if not creds_json:
+        logger.warning("GOOGLE_SHEETS_CREDENTIALS not set")
+        return None
+
+    try:
+        from google.oauth2.service_account import Credentials
+        creds_dict = json.loads(creds_json)
+        scopes = [
+            "https://www.googleapis.com/auth/spreadsheets",
+            "https://www.googleapis.com/auth/drive"
+        ]
+        return Credentials.from_service_account_info(creds_dict, scopes=scopes)
+    except Exception as e:
+        logger.error(f"Failed to load credentials: {e}")
+        return None
+
+
+def _get_spreadsheet():
+    """Get or create the Google Sheets client."""
+    global _spreadsheet
+
+    if _spreadsheet is not None:
+        return _spreadsheet
+
+    try:
+        import gspread
+        creds = _get_credentials()
+        if creds is None:
+            return None
+
+        client = gspread.authorize(creds)
+        _spreadsheet = client.open_by_key(SHEET_ID)
+        logger.info(f"Connected to Google Sheet: {SHEET_ID}")
+        return _spreadsheet
+    except Exception as e:
+        logger.error(f"Failed to connect to Google Sheets: {e}")
+        return None
+
+
+def _get_or_create_worksheet(name: str, headers: List[str]):
+    """Get or create a worksheet with headers."""
+    spreadsheet = _get_spreadsheet()
+    if spreadsheet is None:
+        return None
+
+    try:
+        # Try to get existing worksheet
+        try:
+            worksheet = spreadsheet.worksheet(name)
+        except gspread.WorksheetNotFound:
+            # Create new worksheet
+            worksheet = spreadsheet.add_worksheet(title=name, rows=1000, cols=len(headers))
+            worksheet.append_row(headers)
+            logger.info(f"Created worksheet: {name}")
+
+        return worksheet
+    except Exception as e:
+        logger.error(f"Failed to get/create worksheet {name}: {e}")
+        return None
+
+
+def save_rating_to_sheets(
+    query: str,
+    category: str,
+    entity_id: str,
+    entity_name: str,
+    rank: int,
+    score: float,
+    rating: str,
+    page: int,
+    client_ip: str = "unknown"
+) -> bool:
+    """
+    Save an entity rating to Google Sheets.
+
+    Returns True if successful, False otherwise.
+    """
+    headers = [
+        "timestamp", "client_ip", "query", "category",
+        "entity_id", "entity_name", "rank", "score", "rating", "page"
+    ]
+
+    worksheet = _get_or_create_worksheet(RATINGS_WORKSHEET, headers)
+    if worksheet is None:
+        return False
+
+    try:
+        row = [
+            datetime.now().isoformat(),
+            client_ip,
+            query,
+            category,
+            entity_id,
+            entity_name,
+            rank,
+            round(score, 4) if isinstance(score, float) else score,
+            rating,
+            page
+        ]
+        worksheet.append_row(row)
+        logger.info(f"Saved rating: {entity_name} - {rating}")
+        return True
+    except Exception as e:
+        logger.error(f"Failed to save rating: {e}")
+        return False
+
+
+def save_notes_to_sheets(
+    query: str,
+    category: str,
+    notes: str,
+    client_ip: str = "unknown"
+) -> bool:
+    """
+    Save notes/feedback to Google Sheets.
+
+    Returns True if successful, False otherwise.
+    """
+    headers = ["timestamp", "client_ip", "query", "category", "notes"]
+
+    worksheet = _get_or_create_worksheet(NOTES_WORKSHEET, headers)
+    if worksheet is None:
+        return False
+
+    try:
+        row = [
+            datetime.now().isoformat(),
+            client_ip,
+            query,
+            category,
+            notes
+        ]
+        worksheet.append_row(row)
+        logger.info(f"Saved notes for query: {query[:50]}...")
+        return True
+    except Exception as e:
+        logger.error(f"Failed to save notes: {e}")
+        return False
+
+
+def is_sheets_enabled() -> bool:
+    """Check if Google Sheets storage is configured and working."""
+    return os.environ.get("GOOGLE_SHEETS_CREDENTIALS") is not None
+
+
+def test_connection() -> Dict[str, Any]:
+    """Test the Google Sheets connection."""
+    result = {
+        "configured": is_sheets_enabled(),
+        "connected": False,
+        "sheet_id": SHEET_ID,
+        "error": None
+    }
+
+    if not result["configured"]:
+        result["error"] = "GOOGLE_SHEETS_CREDENTIALS not set"
+        return result
+
+    try:
+        spreadsheet = _get_spreadsheet()
+        if spreadsheet:
+            result["connected"] = True
+            result["title"] = spreadsheet.title
+            result["worksheets"] = [ws.title for ws in spreadsheet.worksheets()]
+    except Exception as e:
+        result["error"] = str(e)
+
+    return result

requirements.txt

@@ -7,3 +7,5 @@ fastapi
 uvicorn[standard]
 python-multipart
 httpx
+gspread>=5.0.0
+google-auth>=2.0.0