Skip to main content

AI-Powered Chat Interface for Jupyter Notebooks

Project description

JupyterWhisper - AI-Powered Chat Interface for Jupyter Notebooks

JupyterWhisper transforms your Jupyter notebook environment by seamlessly integrating Claude AI capabilities. This extension enables natural chat interactions, intelligent code execution, and voice command features to enhance your notebook workflow.

✨ Key Features

  • 🤖 Native integration with Claude 3.5 Sonnet
  • 🎯 Intelligent code execution and cell management
  • 🔍 Advanced search capabilities powered by Perplexity AI
  • 🎙️ Voice command support using OpenAI Whisper
  • 📝 Context-aware text processing and formatting
  • 💬 Comprehensive chat history management
  • ⚡ Real-time streaming responses

🚀 Installation

pip install jupyter_whisper

📋 Requirements

  • Python 3.7+
  • JupyterLab 4.0+ (important: this extension is designed for JupyterLab, not classic Notebook)
  • Jupyter Notebook 7.0+ (if using Notebook instead of Lab)
  • Required API keys:
    • Anthropic API key (for Claude integration)
    • OpenAI API key (optional, for voice features)
    • Perplexity API key (for advanced search capabilities)

Installation Steps

  1. Install JupyterLab if you haven't already:
pip install jupyterlab>=4.0.0
  1. Install Jupyter Whisper:
pip install jupyter_whisper
  1. Start JupyterLab:
jupyter lab

Important Note About Server Management

Jupyter Whisper runs a local FastAPI server (on port 5000) to handle features like audio transcription and text processing. The server is shared between notebooks for efficiency.

Important Notes:

  • The server persists between notebook sessions
  • Configuration changes (like API keys) only take effect when the server restarts
  • You'll be notified if you're using an older server version

To manually refresh the server and apply new configurations:

from jupyter_whisper import refresh_jupyter_whisper
refresh_jupyter_whisper()  # Warning: affects all active notebooks

When to refresh:

  • After updating API keys
  • After upgrading the package
  • If you encounter configuration issues

Note: Refreshing the server will impact all notebooks currently using it. You may need to restart kernels in affected notebooks.

JupyterLab Compatibility

JupyterWhisper is specifically designed and tested for JupyterLab 4.0+. While it may work in classic Jupyter Notebook (7.0+), we recommend using JupyterLab for the best experience and full feature support.

Key compatibility notes:

  • Voice features require a modern browser
  • WebSocket support is required for real-time streaming
  • Some features may require JupyterLab extensions to be enabled
  • Port 5000 must be available for the local server

🔧 Configuration

Interactive Setup

The easiest way to configure Jupyter Whisper is through the interactive setup interface:

import jupyter_whisper

This will open an interactive UI with tabs for:

  • API Keys configuration
  • Model selection
  • System prompt customization

Manual Configuration

You can also configure settings programmatically:

from jupyter_whisper.config import get_config_manager
config = get_config_manager()

# Set API keys
config.set_api_key('ANTHROPIC_API_KEY', 'your-key-here')
config.set_api_key('OPENAI_API_KEY', 'your-key-here')      # Optional for voice
config.set_api_key('PERPLEXITY_API_KEY', 'your-key-here')  # For search

# Change the model
config.set_model('claude-3-5-sonnet-20241022')

# Update system prompt
config.set_system_prompt("Your custom system prompt here")

# Set other preferences
config.set_config_value('SKIP_SETUP_POPUP', True)

Available models:

  • claude-3-5-sonnet-20241022
  • claude-3-5-haiku-20241022
  • claude-3-opus-20240229
  • claude-3-sonnet-20240229
  • claude-3-haiku-20240307

💡 Usage

Basic Chat

Interact with the AI using the %%user magic command:

%%user
How do I read a CSV file using pandas?

Online Search

Access web information directly within your notebook:

from jupyter_whisper import search_online
style = "Be precise and concise"
question = "What's new in Python 3.12?"
search_online(style, question)

Voice Commands

Leverage voice input capabilities:

  • Control recording with keyboard shortcuts
  • Automatic speech-to-text conversion
  • Seamless chat interface integration

🛠️ Advanced Features

Magic Commands

  • %%user [index] - Initiate a user message
  • %%user [index]:set - Replace user message at given index
  • %%assistant [index] - Include assistant response
  • %%assistant [index]:set - Replace assistant message at given index
  • %%assistant [index]:add - Concatenate content to existing assistant message

Example usage:

%%user 3:set
How do I read a CSV file?

%%assistant 3:set
Here's how to read a CSV file using pandas:
import pandas as pd
df = pd.read_csv('file.csv')

%%assistant 3:add
You can also specify additional parameters:
df = pd.read_csv('file.csv', encoding='utf-8')

🔧 Development

Setup Development Environment

git clone https://github.com/yourusername/jupyter_whisper.git
cd jupyter_whisper
pip install -e ".[dev]"

🤝 Contributing

We welcome contributions! Please submit your Pull Requests.

📄 License

MIT License - see LICENSE for details

🙏 Credits

Powered by:


Made with ❤️ by Maxime

Note: This project is independent and not affiliated with Anthropic, OpenAI, or Perplexity AI.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

jupyter_whisper-0.1.5.tar.gz (26.5 kB view details)

Uploaded Source

Built Distribution

jupyter_whisper-0.1.5-py3-none-any.whl (25.8 kB view details)

Uploaded Python 3

File details

Details for the file jupyter_whisper-0.1.5.tar.gz.

File metadata

  • Download URL: jupyter_whisper-0.1.5.tar.gz
  • Upload date:
  • Size: 26.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.12.7

File hashes

Hashes for jupyter_whisper-0.1.5.tar.gz
Algorithm Hash digest
SHA256 34de099dd7ab89bad2a49a6b02008e6a1556a84b20db94c660013247be459bbf
MD5 e3f3a56736e699add9c381fab90fee30
BLAKE2b-256 f2140d7d1fc1e0bcfd522df53b0324762cb2d9f853e76345f4bdce2b7f269784

See more details on using hashes here.

File details

Details for the file jupyter_whisper-0.1.5-py3-none-any.whl.

File metadata

File hashes

Hashes for jupyter_whisper-0.1.5-py3-none-any.whl
Algorithm Hash digest
SHA256 6a8354f5be1a29204c38a3bd85ec405d39457c3573df014adb1c1f34455da8db
MD5 a574cf7d1f44df91a09229b5734e1a82
BLAKE2b-256 a53d1a7d2812fc8d1af1124d7d854f63de56ca859dd0712440e7515dd501acab

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page