Update app/google_oauth.py

app/google_oauth.py

@@ -12,10 +12,11 @@ SCOPES = [
     "https://www.googleapis.com/auth/spreadsheets",
     "https://www.googleapis.com/auth/drive.file",
 ]
+
 TOKEN_FILENAME = "google_token.json"
 DATA_DIR = Path("/data")
 FALLBACK_DATA_DIR = Path(tempfile.gettempdir()) / "tcc2_agent"
-OAUTH_TEMP = {}
+
 
 class GoogleOAuthError(RuntimeError):
     pass
@@ -38,8 +39,7 @@ def _resolve_data_dir():
             continue
 
     raise GoogleOAuthError(
-        "Nao foi possivel criar um diretorio para armazenar o token OAuth. "
-        "Defina GOOGLE_TOKEN_DIR para um caminho gravavel."
+        "Nao foi possivel criar um diretorio para armazenar o token OAuth."
     )
 
 
@@ -61,6 +61,7 @@ def _client_config():
         }.items()
         if not value
     ]
+
     if missing:
         raise GoogleOAuthError("Variaveis OAuth ausentes: " + ", ".join(missing))
 
@@ -70,23 +71,14 @@ def _client_config():
             "client_secret": client_secret,
             "auth_uri": "https://accounts.google.com/o/oauth2/auth",
             "token_uri": "https://oauth2.googleapis.com/token",
-            "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs",
             "redirect_uris": [redirect_uri],
         }
     }
 
 
 def build_flow(state=None):
-    redirect_uri = os.environ.get("GOOGLE_REDIRECT_URI", "").strip()
-
-    flow = Flow.from_client_config(
-        _client_config(),
-        scopes=SCOPES,
-        state=state,
-        autogenerate_code_verifier=False,
-    )
-
-    flow.redirect_uri = redirect_uri
+    flow = Flow.from_client_config(_client_config(), scopes=SCOPES, state=state)
+    flow.redirect_uri = os.environ.get("GOOGLE_REDIRECT_URI")
     return flow
 
 
@@ -99,16 +91,22 @@ def get_authorization_url():
         include_granted_scopes="true",
     )
 
-    return authorization_url, state
+    return authorization_url, state  # importante retornar state
 
 
 def save_credentials(credentials):
+    if not credentials.refresh_token:
+        raise GoogleOAuthError(
+            "Refresh token nao recebido. Revogue o acesso e autorize novamente."
+        )
+
     token_path = get_token_path()
     token_path.write_text(credentials.to_json(), encoding="utf-8")
 
 
 def load_credentials():
     token_path = get_token_path()
+
     if not token_path.exists():
         return None
 
@@ -133,6 +131,7 @@ def load_credentials():
 
 def get_sheets_service():
     credentials = load_credentials()
+
     if credentials is None:
         raise GoogleAuthRequiredError(
             "Google Sheets nao autorizado. Acesse /auth/google primeiro."
@@ -143,33 +142,42 @@ def get_sheets_service():
 
 def append_feedback_to_sheet(feedback_data):
     spreadsheet_id = os.environ.get("GOOGLE_SPREADSHEET_ID", "").strip()
+
     if not spreadsheet_id:
         raise GoogleOAuthError("GOOGLE_SPREADSHEET_ID nao configurado.")
 
     service = get_sheets_service()
+
     fields = [
-    "timestamp",
-    "query",
-    "product_id",
-    "product_name",
-    "rating",
-    "is_helpful",
-    "note",
-    "categoria_inferida",
-    "feedback",
-    "motivo",
-    "score_final",
-    "score_semantico",
-    "bonus_lexical",
-    "penalidade_feedback",
-    "user_message",
-]
+        "timestamp",
+        "search_id",
+        "query",
+        "rank",
+        "product_id",
+        "product_name",
+        "categoria_inferida",
+        "categoria_produto",
+        "rating",
+        "is_helpful",
+        "feedback",
+        "motivo",
+        "score_final",
+        "score_semantico",
+        "bonus_lexical",
+        "penalidade_feedback",
+        "user_message",
+        "note",
+    ]
+
     row = [[str(feedback_data.get(field, "") or "") for field in fields]]
 
-    return service.spreadsheets().values().append(
-        spreadsheetId=spreadsheet_id,
-        range="Feedback!A:Z",
-        valueInputOption="USER_ENTERED",
-        insertDataOption="INSERT_ROWS",
-        body={"values": row},
-    ).execute()
+    try:
+        return service.spreadsheets().values().append(
+            spreadsheetId=spreadsheet_id,
+            range="Feedback!A:Z",
+            valueInputOption="USER_ENTERED",
+            insertDataOption="INSERT_ROWS",
+            body={"values": row},
+        ).execute()
+    except Exception as e:
+        raise GoogleOAuthError(f"Erro ao enviar para Google Sheets: {e}")