Add query_id to Google Sheets and auto-insert headers if missing

backend/api.py

@@ -384,7 +384,8 @@ async def api_entity_feedback(request: EntityFeedbackRequest, req: Request):
                     rating=ratings_str,
                     page=1,
                     client_ip=client_ip,
-                    comment=request.comment or ""
+                    comment=request.comment or "",
+                    query_id=request.query_id
                 )
                 sheets_saved = success
             except Exception as e:

ir/sheets_storage.py

@@ -90,6 +90,12 @@ def _get_or_create_worksheet(name: str, headers: List[str]):
         # Try to get existing worksheet
         try:
             worksheet = spreadsheet.worksheet(name)
+            # Check if headers exist (first row)
+            first_row = worksheet.row_values(1)
+            if not first_row or first_row[0] != headers[0]:
+                # Insert headers at row 1
+                worksheet.insert_row(headers, 1)
+                logger.info(f"Added headers to worksheet: {name}")
         except gspread.WorksheetNotFound:
             # Create new worksheet
             worksheet = spreadsheet.add_worksheet(title=name, rows=1000, cols=len(headers))
@@ -112,7 +118,8 @@ def save_rating_to_sheets(
     rating: str,
     page: int,
     client_ip: str = "unknown",
-    comment: str = ""
+    comment: str = "",
+    query_id: str = ""
 ) -> bool:
     """
     Save an entity rating to Google Sheets.
@@ -120,7 +127,7 @@ def save_rating_to_sheets(
     Returns True if successful, False otherwise.
     """
     headers = [
-        "timestamp", "client_ip", "query", "category",
+        "timestamp", "query_id", "client_ip", "query", "category",
         "entity_id", "entity_name", "rank", "score", "rating", "page", "comment"
     ]
 
@@ -131,6 +138,7 @@ def save_rating_to_sheets(
     try:
         row = [
             datetime.now().isoformat(),
+            query_id,
             client_ip,
             query,
             category,