QuerySUTRA 0.3.3__py3-none-any.whl → 0.4.1__py3-none-any.whl

querysutra-0.4.1.dist-info/METADATA

@@ -0,0 +1,264 @@
+Metadata-Version: 2.4
+Name: QuerySUTRA
+Version: 0.4.1
+Summary: SUTRA: AI-powered data analysis with automatic MySQL export
+Author: Aditya Batta
+License: MIT
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: pandas>=1.3.0
+Requires-Dist: numpy>=1.21.0
+Requires-Dist: openai>=1.0.0
+Requires-Dist: plotly>=5.0.0
+Requires-Dist: matplotlib>=3.3.0
+Requires-Dist: PyPDF2>=3.0.0
+Requires-Dist: python-docx>=0.8.11
+Requires-Dist: openpyxl>=3.0.0
+Provides-Extra: mysql
+Requires-Dist: sqlalchemy>=1.4.0; extra == "mysql"
+Requires-Dist: mysql-connector-python>=8.0.0; extra == "mysql"
+Provides-Extra: postgres
+Requires-Dist: sqlalchemy>=1.4.0; extra == "postgres"
+Requires-Dist: psycopg2-binary>=2.9.0; extra == "postgres"
+Provides-Extra: embeddings
+Requires-Dist: sentence-transformers>=2.0.0; extra == "embeddings"
+Provides-Extra: all
+Requires-Dist: sqlalchemy>=1.4.0; extra == "all"
+Requires-Dist: mysql-connector-python>=8.0.0; extra == "all"
+Requires-Dist: psycopg2-binary>=2.9.0; extra == "all"
+Requires-Dist: sentence-transformers>=2.0.0; extra == "all"
+Dynamic: license-file
+Dynamic: requires-python
+
+# QuerySUTRA
+
+**SUTRA: Structured-Unstructured-Text-Retrieval-Architecture**
+
+AI-powered data analysis library. Upload PDFs, query with natural language, export to MySQL automatically.
+
+## Installation
+
+```bash
+pip install QuerySUTRA
+pip install QuerySUTRA[mysql]  # For MySQL export
+```
+
+## Quick Start
+
+```python
+from sutra import SUTRA
+
+# Upload PDF and auto-export to MySQL in ONE step
+sutra = SUTRA(api_key="your-openai-key")
+
+sutra.upload("data.pdf", auto_export_mysql={
+    'host': 'localhost',
+    'user': 'root', 
+    'password': '123456',
+    'database': 'my_database'  # Auto-creates if not exists
+})
+
+# Query immediately
+result = sutra.ask("Show me all people")
+print(result.data)
+```
+
+## Features
+
+**1. Automatic MySQL Export**
+
+Database auto-created if not exists. No errors.
+
+```python
+# Upload and export to MySQL automatically
+sutra.upload("data.pdf", auto_export_mysql={
+    'host': 'localhost',
+    'user': 'root',
+    'password': 'your_password',
+    'database': 'my_new_database'  # Creates automatically
+})
+```
+
+**2. Complete Data Extraction**
+
+Processes entire PDF in chunks. Extracts ALL employees (not just first 10).
+
+```python
+sutra.upload("large_document.pdf")  # Extracts all 50+ employees
+sutra.tables()  # Shows all extracted tables
+```
+
+**3. Natural Language Queries**
+
+```python
+result = sutra.ask("Show all people from California")
+result = sutra.ask("Who has Python skills?", table="skills")
+result = sutra.ask("Count employees by state", viz="pie")
+```
+
+**4. Custom Visualizations**
+
+```python
+result = sutra.ask("Sales by region", viz="pie")
+result = sutra.ask("Trends", viz="line")
+result = sutra.ask("Compare", viz="bar")
+result = sutra.ask("Data", viz="scatter")
+```
+
+**5. Load Existing Databases**
+
+```python
+# Load SQLite
+sutra = SUTRA.load_from_db("data.db", api_key="key")
+
+# Connect to MySQL
+sutra = SUTRA.connect_mysql("localhost", "root", "pass", "database")
+
+# Connect to PostgreSQL  
+sutra = SUTRA.connect_postgres("localhost", "postgres", "pass", "database")
+```
+
+**6. Smart Features (Optional)**
+
+```python
+sutra = SUTRA(
+    api_key="your-key",
+    use_embeddings=True,    # Cache similar queries (saves API calls)
+    fuzzy_match=True,       # "New York City" matches "New York"
+    check_relevance=True,   # Detect irrelevant queries
+    cache_queries=True      # Cache exact queries
+)
+```
+
+**7. Direct SQL (Free)**
+
+```python
+result = sutra.sql("SELECT * FROM people WHERE state='CA'")
+print(result.data)
+```
+
+## Complete Workflow
+
+**In Colab:**
+```python
+from sutra import SUTRA
+
+sutra = SUTRA(api_key="your-key")
+sutra.upload("employee_data.pdf")
+sutra.tables()  # See extracted tables
+
+# Export and download
+sutra.export_db("data.db", format="sqlite")
+from google.colab import files
+files.download("data.db")
+```
+
+**On Windows:**
+```python
+from sutra import SUTRA
+
+# Load downloaded database
+sutra = SUTRA.load_from_db("data.db", api_key="your-key")
+
+# Export to MySQL (auto-creates database)
+sutra.save_to_mysql("localhost", "root", "password", "my_database")
+
+# Verify in MySQL
+sutra_mysql = SUTRA.connect_mysql("localhost", "root", "password", "my_database")
+sutra_mysql.tables()
+```
+
+## Export Options
+
+```python
+# SQLite
+sutra.export_db("backup.db", format="sqlite")
+
+# SQL dump
+sutra.export_db("schema.sql", format="sql")
+
+# JSON
+sutra.export_db("data.json", format="json")
+
+# Excel
+sutra.export_db("data.xlsx", format="excel")
+
+# MySQL (auto-creates database)
+sutra.save_to_mysql("localhost", "root", "pass", "new_db")
+
+# PostgreSQL
+sutra.save_to_postgres("localhost", "postgres", "pass", "new_db")
+```
+
+## API Reference
+
+**Initialize**
+```python
+SUTRA(api_key, db, use_embeddings, check_relevance, fuzzy_match, cache_queries)
+```
+
+**Class Methods**
+- `load_from_db(path, api_key)` - Load SQLite
+- `connect_mysql(host, user, password, database)` - Connect MySQL
+- `connect_postgres(host, user, password, database)` - Connect PostgreSQL
+
+**Instance Methods**
+- `upload(data, name, auto_export_mysql)` - Upload with optional auto-export
+- `ask(question, viz, table)` - Natural language query
+- `sql(query, viz)` - Direct SQL
+- `tables()` - List tables
+- `schema(table)` - Show schema
+- `peek(table, n)` - Preview data
+- `export_db(path, format)` - Export database
+- `save_to_mysql(host, user, password, database)` - Export to MySQL (auto-creates DB)
+- `save_to_postgres(...)` - Export to PostgreSQL
+- `backup(path)` - Backup
+- `close()` - Close
+
+## Troubleshooting
+
+**MySQL database doesn't exist**
+- Fixed in v0.4.0 - auto-creates database automatically
+- No need to manually create database
+
+**Only 10 employees extracted from 50-employee PDF**
+- Fixed in v0.4.0 - processes entire PDF in chunks
+- Upgrade: `pip install --upgrade QuerySUTRA`
+
+**connect_mysql() not found**
+- Update: `pip install --upgrade QuerySUTRA`
+- Install MySQL support: `pip install QuerySUTRA[mysql]`
+
+## Supported Formats
+
+CSV, Excel, JSON, SQL, PDF, Word, Text, Pandas DataFrame
+
+## Requirements
+
+- Python 3.8+
+- OpenAI API key
+- MySQL/PostgreSQL (optional)
+
+## License
+
+MIT License
+
+## Changelog
+
+**v0.4.0**
+- AUTO-CREATES MySQL database (no more errors)
+- Complete PDF extraction (all pages, all employees)
+- Chunk processing for large documents
+- One-line auto-export to MySQL
+- Simplified everything
+
+**v0.3.x**
+- MySQL/PostgreSQL connectivity
+- Embeddings caching
+- Fuzzy matching
+- Custom visualizations
+
+---
+
+**Made by Aditya Batta**

{querysutra-0.3.3.dist-info → querysutra-0.4.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
-querysutra-0.3.3.dist-info/licenses/LICENSE,sha256=F-4b93u0OVrVwGXgMwBRq6MlGyUT9zmre1oh5Gft5Ts,1066
-sutra/__init__.py,sha256=ujgmobOPa4UHLpXv7Nz84qQ2B2_BOHfyNE0lW1TlDuw,335
+querysutra-0.4.1.dist-info/licenses/LICENSE,sha256=F-4b93u0OVrVwGXgMwBRq6MlGyUT9zmre1oh5Gft5Ts,1066
+sutra/__init__.py,sha256=kGYTlJ8CEoYHF47rhU41kBzoNu7MblOHYSABe2cZsFc,178
 sutra/cache_manager.py,sha256=e0AAeUqoR-aiqzZ3fB-IDvpJ4JA6-YBFyRJxusEnIrA,3082
 sutra/clear_cache.py,sha256=rVIz29p7V11Uh6oHXeaWpFtYXXv-2OED91cHMAWWxtQ,187
 sutra/core.py,sha256=R_JbOlZTukegP92Dr-WLsdr632_otFN7o9qSvcxyBtw,10497
@@ -11,7 +11,7 @@ sutra/feedback_matcher.py,sha256=WXYpGtFJnOyYQOzy-z8uBiUWH5vyJJOMS1NwEYzNfic,286
 sutra/nlp_processor.py,sha256=wMS1hz1aGWjSwPUD7lSNBbQapFtLgF2l65j0QKXQOd0,5461
 sutra/schema_embeddings.py,sha256=bVPzpJOdYTyUdG2k3ZdgYJLrX2opHBx68RIjJcMlueo,9732
 sutra/schema_generator.py,sha256=BX_vXmnvSGc6nCBx40WLSoNL3WIYPDahd1cEYloyY4M,1925
-sutra/sutra.py,sha256=EGMBebWAYLzwADozUNbMQfEPv4xGCtL5zsbymPJ43HU,40554
+sutra/sutra.py,sha256=GQ-3gLKoDsSDAgb57Yfc7uHFdx_ptYryxCFS8Jr0Nmc,33082
 sutra/sutra_client.py,sha256=PYYDGqVbA9pB-Zcsm52i9KarwijCIGVZOThgONZP6Vs,14203
 sutra/sutra_core.py,sha256=diaWOXUHn1wrqCQrBhLKL612tMQioaqx-ILc3y9-CqM,11708
 sutra/sutra_simple.py,sha256=rnqzG7OAt4p64XtO0peMqHS1pG5tdA8U3EYTMVsq7BE,23201
@@ -22,7 +22,7 @@ tests/test_sutra.py,sha256=6Z4SoIuBzza101304I7plkyPVkUBbjIxR8uPs9z5ntg,2383
 utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 utils/file_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 utils/text_utils.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-querysutra-0.3.3.dist-info/METADATA,sha256=DRUhb00cXQrf9JIjfbp9CiSJkoPrlgsY1wmWWjQ1pdI,7637
-querysutra-0.3.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-querysutra-0.3.3.dist-info/top_level.txt,sha256=9v0buw21eo5LaUU_3Cf9b9MqRyEvtM9cHaOuEXUKVqM,18
-querysutra-0.3.3.dist-info/RECORD,,
+querysutra-0.4.1.dist-info/METADATA,sha256=w3LFyZF2vZ_Ne4ZKcDRPTHBRWK9HOPjOBjnNMCrQGHk,6792
+querysutra-0.4.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+querysutra-0.4.1.dist-info/top_level.txt,sha256=9v0buw21eo5LaUU_3Cf9b9MqRyEvtM9cHaOuEXUKVqM,18
+querysutra-0.4.1.dist-info/RECORD,,

sutra/__init__.py

@@ -1,11 +1,8 @@
 """
-QuerySUTRA - Structured-Unstructured-Text-Retrieval-Architecture
-Creates multiple structured tables from ANY data with AI
-
-v0.3.3 - FIXED: Proper primary/foreign keys and comprehensive extraction
+QuerySUTRA v0.4.1 - Simple & Automatic
 """
 
-__version__ = "0.3.3"
+__version__ = "0.4.1"
 
 from sutra.sutra import SUTRA, QueryResult, quick_start