Metadata-Version: 2.4
Name: ariai
Version: 0.1.0
Summary: A flexible chatbot framework supporting multiple AI providers
Home-page: https://github.com/yourusername/ariai
Author: AriAI Team
Author-email: your.email@example.com
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Requires-Python: >=3.8
Description-Content-Type: text/markdown
Requires-Dist: openai>=0.27.0
Requires-Dist: transformers>=4.30.0
Requires-Dist: torch>=2.0.0
Requires-Dist: rich>=13.0.0
Requires-Dist: gradio>=4.0.0
Dynamic: author
Dynamic: author-email
Dynamic: classifier
Dynamic: description
Dynamic: description-content-type
Dynamic: home-page
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary

# AriAI

A flexible chatbot framework that supports multiple AI providers (OpenAI and Hugging Face).

## Features

- Support for both OpenAI (GPT-3.5/4) and Hugging Face models
- Custom system prompts / personalities
- Conversation memory
- Rich terminal UI with colored output
- Easy to use and extend

## Installation

1. Clone this repository:
```bash
git clone https://github.com/username/ariai.git
cd ariai
```

2. Build and install the package:
```bash
python setup.py sdist bdist_wheel
pip install ./dist/ariai-0.1-py3-none-any.whl
```

## Usage

Here's a basic example:

```python
from ariai import AriAI

# Using OpenAI
bot = AriAI(
    provider="openai",
    api_key="YOUR_OPENAI_API_KEY",
    system_prompt="You are a helpful AI assistant."
)

# Or using Hugging Face
bot = AriAI(
    provider="huggingface",
    hf_model="google/flan-t5-base",
    system_prompt="You are a helpful AI assistant."
)

# Start the interactive chat UI
bot.start_ui()
```

You can also use the chat method directly:

```python
response = bot.chat("Hello, how are you?")
print(response)
```

## Configuration

### OpenAI

For OpenAI, you'll need to set your API key. You can either:
1. Pass it directly to the constructor
2. Set it as an environment variable: `OPENAI_API_KEY`

### Hugging Face

For Hugging Face, you can specify any model that supports text generation. The default is "google/flan-t5-base".

## Contributing

Contributions are welcome! Please feel free to submit a Pull Request.

## License

This project is licensed under the MIT License - see the LICENSE file for details.
