Google AI Studio Pro Enhancer

A powerful Tampermonkey/Greasemonkey script designed to enhance the quality of life and functionality within the Google AI Studio interface. It adds advanced navigation, library organization, detailed metadata, and customization options.

✨ Visual Tour & Features

1. Enhanced Library Management

The library view is completely transformed to provide more data at a glance.

• New Columns: Adds "Created" (Date), "Count" (Messages), and "Tokens" (Total context).
• Visual Grouping: Automatically detects "Branch of" and "Copy of" relationships, visually indenting them under their parent chat for a clean file tree structure.
• Sorting: All new columns are sortable.

2. Family Tree Toolbar & Session Metadata

Never lose track of your conversation branches again.

• Family Tree: The top toolbar displays numbered links to the Parent, Siblings, and Children of the current chat.
• Smart Coloring: Links turn Green if that branch is newer than the one you are currently viewing.
• Session Info: Displays the exact creation date and a relative "Last Modified" timer right next to the token count.

3. Smart Navigation Tools

Navigate long context windows with ease using dedicated tools injected into the UI.

• Find in Chat: A custom search bar (accessible via the 3-dot menu) allows you to search specifically within the chat message content, highlighting results and jumping between them.
• Dual Scroll Buttons: Floating Action Buttons (FABs) in the bottom right allow you to jump precisely to the Previous or Next message turn, skipping over long blocks of text.

Find in Chat	Scroll Navigator

4. Customization & Data Sync

A new "Enhancer" button in the Settings menu opens a dedicated configuration modal.

• Appearance: Customize date formats (Short, American, Relative) and the color of the Token count in the library.
• Force Sync: A dedicated button to force a synchronization with Google Drive, useful for cleaning up "ghost" chats or updating metadata immediately.

5. Toolbar Cleanup & Export

To reduce clutter, native buttons like "Share", "Compare", and "New Chat" are moved into the "More Actions" (3-dot) menu.

• Export Options: Adds one-click options to export your full chat history to JSON or Markdown.

---

🔑 Google Credentials Configuration

The script requires Google Drive API access to fetch creation and modification dates for your conversations. You need to generate a "Refresh Token" so the script can stay logged in.

Step 1: Google Cloud Setup (Required for both methods)

1. Go to the Google Cloud Console.
2. Create a new project (e.g., named "AI Studio Enhancer").
3. Enable the Drive API:
   • Go to APIs & Services > Library.
   • Search for "Google Drive API" and click Enable.
4. Create Credentials:
   • Go to APIs & Services > Credentials.
   • Click Create Credentials > OAuth 2.0 Client ID.
   • (If prompted, configure the Consent Screen: Select "External", give it a name, and save).
   • Application Type: Select Desktop app.
   • Click Create.

---

Step 2: Generate Tokens (Choose one method)

🅰️ Option A: Python Script (Recommended/Easy)

We have provided a script that automates the process.

1. Download JSON: In the Google Cloud Console (where you created credentials in Step 1), click the Download JSON button for your OAuth Client.
2. Save File: Save the file into the same folder as this repo.
   • Note: The filename will start with client_secret_... and end in .json.
3. Install Requirements:

   pip install google-auth-oauthlib

4. Run Script:

   python get_drive_tokens.py

5. Follow Prompts: The script will detect your JSON file, give you a URL to visit to authorize, and then output the exact configuration block you need.

🅱️ Option B: OAuth Playground (Manual/No Code)

If you don't want to run Python, you can do this manually.

1. Get ID & Secret: In Google Cloud Console, copy your Client ID and Client Secret.
2. Go to the OAuth 2.0 Playground.
3. Click the Settings (Gear icon) in the top right.
   • Check Use your own OAuth credentials.
   • Paste your Client ID and Client Secret.
4. Select Scopes:
   • In the list on the left, find "Drive API v3".
   • Select https://www.googleapis.com/auth/drive.readonly.
   • Click Authorize APIs.
5. Exchange Tokens:
   • Click "Exchange authorization code for tokens".
   • Copy the Refresh Token from the response.

---

Step 3: Update the UserScript

Whichever method you used, you now have a Client ID, Client Secret, and Refresh Token.

1. Open the aistudio-plus.user.js script in your Tampermonkey/Greasemonkey editor.
2. Find the CONFIG block at the top of the file:

const CONFIG = {
    CLIENT_ID: "YOUR_CLIENT_ID",
    CLIENT_SECRET: "YOUR_CLIENT_SECRET",
    REFRESH_TOKEN: "YOUR_REFRESH_TOKEN",
    FOLDER_NAME: "Google AI Studio",
    SYNC_INTERVAL_MINUTES: 15
};

3. Replace the placeholder values with your actual credentials.
4. Save and reload AI Studio.

---

💾 Installation

1. Install Manager: Get Tampermonkey (Chrome/Edge) or Greasemonkey (Firefox).
2. Install Script: Click here to install aistudio-plus.user.js
3. Configure: Follow the "Google Credentials Configuration" steps above.
4. Enjoy: Reload Google AI Studio.

📜 Development

Developed by Esashiero, with architectural guidance provided by Gemini.

---

Developed for the Google AI Studio platform.