dracula-ai 0.3.6

A simple Python library for Google Gemini with memory support.

Dracula 🧛

A simple, elegant Python library for Google Gemini with powerful features.

Installation

pip install dracula-ai

Quick Start

from dracula import Dracula
from dotenv import load_dotenv
import os

load_dotenv()

ai = Dracula(api_key=os.getenv("GEMINI_API_KEY"))
response = ai.chat("Hello, who are you?")
print(response)

Parameters

Parameter	Type	Default	Description
api_key	str	required	Your Google Gemini API key
model	str	gemini-2.0-flash	Gemini model name
max_messages	int	10	Maximum number of messages to remember
prompt	str	"You are a helpful assistant."	System prompt
temperature	float	1.0	Response creativity (0.0 - 2.0)
max_output_tokens	int	8192	Maximum response length
stats_filepath	str	"dracula_stats.json"	Path to save usage stats
language	str	"English"	Language for responses

Features

💬 Text Chat

ai = Dracula(api_key="your-api-key")
response = ai.chat("What is Python?")
print(response)

🌊 Streaming

for chunk in ai.stream("Tell me a long story."):
    print(chunk, end="", flush=True)

🧠 Conversation Memory

ai.chat("My name is Ahmet.")
response = ai.chat("What is my name?")
print(response)  # It remembers! ✅

ai.clear_memory()  # Wipe memory

💾 Save & Load History

ai.save_history("conversation.json")

# Later:
ai.load_history("conversation.json")

📜 Pretty Print History

ai.print_history()

🎭 System Prompt

ai = Dracula(
    api_key="your-api-key",
    prompt="You are a pirate who answers everything dramatically."
)

🌡️ Temperature Control

ai = Dracula(api_key="your-api-key", temperature=0.2)  # Focused
ai = Dracula(api_key="your-api-key", temperature=1.8)  # Creative

🌍 Response Language

ai = Dracula(api_key="your-api-key", language="Turkish")
response = ai.chat("Hello!")
print(response)  # Merhaba! ✅

📊 Usage Stats

print(ai.get_stats())
# {
#   "total_messages": 5,
#   "total_responses": 5,
#   "total_characters_sent": 120,
#   "total_characters_received": 3400
# }

ai.reset_stats()

🔗 Chainable Methods

ai.set_prompt("You are a chef.").set_temperature(0.9).set_language("Turkish")

🧹 Context Manager

with Dracula(api_key="your-api-key") as ai:
    ai.chat("Hello!")
    ai.print_history()
# Memory and stats automatically reset here ✅

Error Handling

from dracula import ValidationException, ChatException, InvalidAPIKeyException

try:
    ai = Dracula(api_key="", temperature=5.0)
except ValidationException as e:
    print(f"Validation error: {e}")
except InvalidAPIKeyException as e:
    print(f"API key error: {e}")
except ChatException as e:
    print(f"Chat error: {e}")

Getting Your API Key

1. Go to https://aistudio.google.com
2. Sign in with your Google account
3. Click "Get API Key"
4. Copy your key and store it safely in a .env file

License

MIT License — feel free to use this in your own projects!

Author

Suleyman Ibis