Skip to content

Repository files navigation

🚀 LocalLab: Your Personal AI Lab

LocalLab ServerLocalLab ClientLicensePython

Run ChatGPT-like AI on your own computer! LocalLab is a complete AI platform that runs models locally with a powerful chat interface and Python client.

✨ What Makes LocalLab Special?

LocalLab gives you your own personal ChatGPT that runs entirely on your computer:

  • 🎯 Terminal Chat Interface - ChatGPT-like experience in your terminal
  • 🔒 Complete Privacy - Your data never leaves your computer
  • 💰 Zero Cost - No monthly fees or API charges
  • 🌐 Access Anywhere - Use from any device with ngrok tunneling
  • Multiple Models - Support for various open-source AI models
  • 🤖 Model Management - Download, organize, and manage AI models locally
  • 🎮 Free GPU - Run on Google Colab for free GPU acceleration

Perfect for developers, students, researchers, or anyone who wants to experiment with AI without privacy concerns or ongoing costs.

🚀 Quick Start (3 Steps)

# 1. Install LocalLab
pip install locallab locallab-client
# 2. Start your AI server
locallab start
# 3. Chat with your AI
locallab chat

That's it! You now have your own ChatGPT running locally.

🤖 Model Management (Optional)

Want to download models ahead of time or manage your local AI models? LocalLab includes powerful model management:

# Discover available models
locallab models discover
# Download a model locally (faster startup)
locallab models download microsoft/phi-2
# List your cached models
locallab models list
# Get detailed model information
locallab models info microsoft/phi-2

📖 Learn More: See the Model Management Guide for complete documentation.

🧠 How LocalLab Works

LocalLab has three main components:

1. 🖥️ LocalLab Server (pip install locallab)

  • Runs AI models on your computer
  • Provides a web API for interactions
  • Handles model loading and optimization
  • Start with: locallab start

2. 💬 Chat Interface (Built-in)

  • Terminal-based ChatGPT-like experience
  • Real-time streaming responses
  • Multiple generation modes
  • Access with: locallab chat

3. 🤖 Model Management (Built-in)

  • Download and organize AI models locally
  • Discover available models from HuggingFace Hub
  • Manage disk space and cache cleanup
  • Use with: locallab models

4. 🐍 Python Client (pip install locallab-client)

  • Programmatic access for your code
  • Both sync and async support
  • Use with: client = SyncLocalLabClient("http://localhost:8000")
graph TD
A[Terminal Chat] -->|Uses| C[LocalLab Server]
B[Python Code] -->|Uses| C
C -->|Runs| D[AI Models]
C -->|Optional| E[Ngrok Tunnel]
E -->|Access from| F[Any Device]
style C fill:#2563eb,stroke:#1e40af,stroke-width:2px,color:#ffffff
style D fill:#059669,stroke:#047857,stroke-width:2px,color:#ffffff
style A fill:#7c3aed,stroke:#6d28d9,stroke-width:2px,color:#ffffff
style B fill:#dc2626,stroke:#b91c1c,stroke-width:2px,color:#ffffff
style E fill:#ea580c,stroke:#c2410c,stroke-width:2px,color:#ffffff
style F fill:#0891b2,stroke:#0e7490,stroke-width:2px,color:#ffffff
Loading

🌟 The Magic: Use --use-ngrok to access your AI from anywhere - your phone, another computer, or share with friends!

🎯 Key Features

📦 Easy Setup 🔒 Privacy First 🎮 Free GPU Access
🤖 Multiple Models 💾 Memory Efficient 🔄 Auto-Optimization
🗂️ Model Management ⚡ Fast Response 🔧 Simple Server
🌐 Local or Colab 🔌 Client Package 🛡️ Secure Tunneling
🌍 Access Anywhere 📥 Offline Models 🧹 Cache Cleanup

Two-Part System:

  • LocalLab Server: Runs the AI models and exposes API endpoints
  • LocalLab Client: A separate Python package (pip install locallab-client) that connects to the server

Access From Anywhere: With built-in ngrok integration, you can securely access your LocalLab server from any device, anywhere in the world - perfect for teams, remote work, or accessing your models on the go.

🌟 Two Ways to Run

  1. On Your Computer (Local Mode)

    💻 Your Computer
    └── 🚀 LocalLab Server
    └── 🤖 AI Model
    └── 🔧 Auto-optimization
    
  2. On Google Colab (Free GPU Mode)

    ☁️ Google Colab
    └── 🎮 Free GPU
    └── 🚀 LocalLab Server
    └── 🤖 AI Model
    └── ⚡ GPU Acceleration
    

📦 Installation & Setup

Latest Package Versions:

  • LocalLab Server: LocalLab Server
  • LocalLab Client: LocalLab Client

Windows Setup

  1. Install Required Build Tools

  2. Install Packages

    pip install locallab locallab-client
  3. Verify PATH

    • If locallab command isn't found, add Python Scripts to PATH:
      # Find Python locationwhere python
      # This will show something like: C:\Users\YourName\AppData\Local\Programs\Python\Python311\python.exe

    Adding to PATH in Windows:

    1. Press Win + X and select "System"
    2. Click "Advanced system settings" on the right
    3. Click "Environment Variables" button
    4. Under "System variables", find and select "Path", then click "Edit"
    5. Click "New" and add your Python Scripts path (e.g., C:\Users\YourName\AppData\Local\Programs\Python\Python311\Scripts\)
    6. Click "OK" on all dialogs
    7. Restart your command prompt
    • Alternatively, use: python -m locallab start

🔍 Having issues? See our Windows Troubleshooting Guide

Linux/Mac Setup

# Install both server and client packages
pip install locallab locallab-client

2. Configure the Server (Recommended)

# Run interactive configuration
locallab config
# This will help you set up:# - Model selection# - Memory optimizations# - GPU settings# - System resources

3. Start the Server

# Start with saved configuration
locallab start
# Or start with specific options
locallab start --model microsoft/phi-2 --quantize --quantize-type int8

💬 Terminal Chat Interface - Your Personal ChatGPT

The LocalLab Chat Interface is a powerful terminal-based tool that gives you a ChatGPT-like experience right in your command line. It's the easiest way to interact with your AI models.

🎯 Why Use the Chat Interface?

  • Instant AI Access - No coding required, just type and chat
  • Real-time Responses - See AI responses as they're generated
  • Rich Formatting - Markdown rendering with syntax highlighting
  • Smart Features - History, saving, batch processing, and more
  • Works Everywhere - Local, remote, or Google Colab

🚀 Getting Started

# Start your server
locallab start
# Open chat interface
locallab chat

✨ Key Features

FeatureDescriptionExample
Dynamic Mode SwitchingChange generation mode per messageExplain AI --stream
Real-time StreamingSee responses as they're typedLive text generation
Conversation HistoryTrack and save your chats/history, /save
Batch ProcessingProcess multiple prompts/batch command
Remote AccessConnect to any LocalLab server--url https://your-server.com
Error RecoveryAuto-reconnection and graceful handlingSeamless experience

🎮 Interactive Commands

/help # Show all available commands
/history # View conversation history
/save # Save current conversation
/batch # Enter batch processing mode
/reset # Clear conversation history
/exit # Exit gracefully

🔄 Dynamic Mode Switching (New!)

Override the default generation mode for any message:

You: Write a story --stream # Use streaming mode
🔄 Using stream mode for this message
You: Remember my name is Alice --chat # Use chat mode with context
🔄 Using chat mode for this message
You: What's 2+2? --simple # Use simple mode🔄 Using simple mode for this message

📱 Example Chat Session

$ locallab chat
🚀 LocalLab Chat Interface
✅ Connected to: http://localhost:8000
📊 Server: LocalLab v0.9.0 | Model: qwen-0.5b
You: Hello! Can you help me with Python?
AI: Hello! I'd be happy to help you with Python programming.What specific topic would you like to explore?You: Show me how to create a class --streamAI: Here's how to create a simple class in Python:
```pythonclass Person: def __init__(self, name, age): self.name = name self.age = age def introduce(self):return f"Hi, I'm {self.name} and I'm {self.age} years old."# Usageperson = Person("Alice", 25)print(person.introduce())You: /save💾 Conversation saved to: chat_2024-07-06_14-30-15.jsonYou: /exit👋 Goodbye!

🌐 Remote Access

Connect to any LocalLab server from anywhere:

# Connect to remote server
locallab chat --url https://abc123.ngrok.app
# Use with Google Colab
locallab chat --url https://your-colab-ngrok-url.app

📖 Complete Guide: See the Chat Interface Documentation for advanced features, examples, and troubleshooting.

🐍 Python Client - Programmatic Access

For developers who want to integrate AI into their applications, LocalLab provides a powerful Python client package.

🎯 Two Ways to Use LocalLab

MethodBest ForGetting Started
Chat InterfaceInteractive use, testing, quick questionslocallab chat
Python ClientApplications, scripts, automationfrom locallab_client import SyncLocalLabClient

📦 Synchronous Client (Recommended for Beginners)

fromlocallab_clientimportSyncLocalLabClient# Connect to server - choose ONE of these options:# 1. For local server (default)client=SyncLocalLabClient("http://localhost:8000")
# 2. For remote server via ngrok (when using Google Colab or --use-ngrok)# client = SyncLocalLabClient("https://abc123.ngrok.app") # Replace with your ngrok URLtry:
print("Generating text...")
# Generate textresponse=client.generate("Write a story")
print(response)
print("Streaming responses...")
# Stream responsesfortokeninclient.stream_generate("Tell me a story"):
print(token, end="", flush=True)
print("Chat responses...")
# Chat with AIresponse=client.chat([
{"role": "system", "content": "You are helpful."},
{"role": "user", "content": "Hello!"}
])
print(response.choices[0]["message"]["content"])
finally:
# Always close the clientclient.close()

💡 Important: When connecting to a server running on Google Colab or with ngrok enabled, always use the ngrok URL (https://abc123.ngrok.app) that was displayed when you started the server.

Asynchronous Client Usage (For Advanced Users)

importasynciofromlocallab_clientimportLocalLabClientasyncdefmain():
# Connect to server - choose ONE of these options:# 1. For local server (default)client=LocalLabClient("http://localhost:8000")
# 2. For remote server via ngrok (when using Google Colab or --use-ngrok)# client = LocalLabClient("https://abc123.ngrok.app") # Replace with your ngrok URLtry:
print("Generating text...")
# Generate textresponse=awaitclient.generate("Write a story")
print(response)
print("Streaming responses...")
# Stream responsesasyncfortokeninclient.stream_generate("Tell me a story"):
print(token, end="", flush=True)
print("\nChatting with AI...")
# Chat with AIresponse=awaitclient.chat([
{"role": "system", "content": "You are helpful."},
{"role": "user", "content": "Hello!"}
])
# Extracting Contentcontent=response['choices'][0]['message']['content']
print(content)
finally:
# Always close the clientawaitclient.close()
# Run the async functionasyncio.run(main())

🌐 Google Colab Usage with Remote Access

Step 1: Set Up the Server on Google Colab

First, you'll set up the LocalLab server on Google Colab to use their free GPU:

# In your Colab notebook:# 1. Install the server package
!pipinstalllocallab# 2. Configure with CLI (notice the ! prefix)
!locallabconfig# 3. Start server with ngrok for remote access
!locallabstart--use-ngrok# The server will display a public URL like:# 🚀 Ngrok Public URL: https://abc123.ngrok.app# COPY THIS URL - you'll need it to connect!

Step 2: Connect to Your Server

After setting up your server on Google Colab, you'll need to connect to it using the LocalLab client package. The server will display a ngrok URL that you'll use for the connection.

Using the Client Connection Examples

You can now use the client connection examples from the Client Connection & Usage section above.

Just make sure to:

  1. Use your ngrok URL instead of localhost
  2. Install the client package if needed

For example:

# In another cell in the same Colab notebook:# 1. Install the client package
!pipinstalllocallab-client# 2. Import the clientfromlocallab_clientimportSyncLocalLabClient# 3. Connect to your ngrok URL (replace with your actual URL from Step 1)client=SyncLocalLabClient("https://abc123.ngrok.app") # ← REPLACE THIS with your URL!# 4. Now you can use any of the client methodsresponse=client.generate("Write a poem about AI")
print(response)
# 5. Always close when doneclient.close()

Access From Any Device

The power of using ngrok is that you can connect to your Colab server from anywhere:

# On your local computer, phone, or any device with Python:pipinstalllocallab-clientfromlocallab_clientimportSyncLocalLabClientclient=SyncLocalLabClient("https://abc123.ngrok.app") # ← REPLACE THIS with your URL!response=client.generate("Hello from my device!")
print(response)
client.close()

💡 Remote Access Tip: The ngrok URL lets you access your LocalLab server from any device - your phone, tablet, another computer, or share with teammates. See the Client Connection & Usage section above for more examples of what you can do with the client.

💻 Requirements

Local Computer

  • Python 3.8+
  • 4GB RAM minimum (8GB+ recommended)
  • GPU optional but recommended
  • Internet connection for downloading models

Google Colab

  • Just a Google account!
  • Free tier works fine

🌟 Features

  • Easy Setup: Just pip install and run
  • Multiple Models: Use any Hugging Face model
  • Resource Efficient: Automatic optimization
  • Privacy First: All local, no data sent to cloud
  • Free GPU: Google Colab integration
  • Flexible Client API: Both async and sync clients available
  • Automatic Resource Management: Sessions close automatically
  • Remote Access: Access your models from anywhere with ngrok integration
  • Secure Tunneling: Share your models securely with teammates or access from mobile devices
  • Client Libraries: Python libraries for both synchronous and asynchronous usage

🌍 Client-Server Architecture

graph LR
A[Your Application] -->|Uses| B[LocalLab Client]
B -->|API Requests| C[LocalLab Server]
C -->|Runs| D[AI Models]
C -->|Optional| E[Ngrok Tunnel]
E -->|Remote Access| F[Any Device, Anywhere]
style A fill:#7c3aed,stroke:#6d28d9,stroke-width:2px,color:#ffffff
style B fill:#dc2626,stroke:#b91c1c,stroke-width:2px,color:#ffffff
style C fill:#2563eb,stroke:#1e40af,stroke-width:2px,color:#ffffff
style D fill:#059669,stroke:#047857,stroke-width:2px,color:#ffffff
style E fill:#ea580c,stroke:#c2410c,stroke-width:2px,color:#ffffff
style F fill:#0891b2,stroke:#0e7490,stroke-width:2px,color:#ffffff
Loading

➡️ See All Features

📚 Documentation

🚀 Getting Started

GuideDescription
Installation & SetupComplete installation guide for all platforms
CLI OverviewCommand-line interface documentation
Chat InterfaceTerminal chat features and examples

💻 Using LocalLab

GuideDescription
CLI ReferenceComplete command documentation
Model ManagementDownload and organize AI models
Python ClientProgrammatic access guide
API ReferenceHTTP API documentation

🌐 Deployment & Advanced

GuideDescription
Google Colab SetupFree GPU deployment guide
TroubleshootingCommon issues and solutions
Advanced FeaturesPower user features

🔍 Need Help?

📖 Additional Resources

🌟 Star Us!

If you find LocalLab helpful, please star our repository! It helps others discover the project.


Made with ❤️ by Utkarsh Tiwari GitHubTwitterLinkedIn

About

LocalLab allows you to easily run Hugging Face AI models locally or on Google Colab, featuring automatic API setup, model management, performance optimization, and system monitoring.

Topics

Resources

Contributing

Stars

5 stars

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages