athena-lab1 0.0.1__tar.gz

athena_lab1-0.0.1/PKG-INFO

@@ -0,0 +1,28 @@
+Metadata-Version: 2.4
+Name: athena_lab1
+Version: 0.0.1
+Summary: Auto-detects datasets and answers questions via OpenRouter.
+Author-email: Sammy <your.email@example.com>
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.9
+Requires-Dist: openai
+Requires-Dist: pandas
+Requires-Dist: python-dotenv
+Description-Content-Type: text/markdown
+
+# Athena Lab1
+
+A simple package that auto-detects datasets and answers questions using OpenRouter.
+
+## Installation
+
+pip install athena_lab1
+
+## Usage
+
+import pandas as pd
+from athena_lab1.core import ask
+
+dataset = pd.read_csv("mydataset.csv")
+ask("Build a machine learning model")

athena_lab1-0.0.1/README.md

@@ -0,0 +1,15 @@
+# Athena Lab1
+
+A simple package that auto-detects datasets and answers questions using OpenRouter.
+
+## Installation
+
+pip install athena_lab1
+
+## Usage
+
+import pandas as pd
+from athena_lab1.core import ask
+
+dataset = pd.read_csv("mydataset.csv")
+ask("Build a machine learning model")

athena_lab1-0.0.1/pyproject.toml

@@ -0,0 +1,22 @@
+[build-system]
+requires = ["hatchling >= 1.26"]
+build-backend = "hatchling.build"
+
+[project]
+name = "athena_lab1"
+version = "0.0.1"
+authors = [
+  { name="Sammy", email="your.email@example.com" },
+]
+description = "Auto-detects datasets and answers questions via OpenRouter."
+readme = "README.md"
+requires-python = ">=3.9"
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "Operating System :: OS Independent",
+]
+dependencies = [
+    "openai",
+    "pandas",
+    "python-dotenv",
+]

athena_lab1-0.0.1/src/athena_lab1/__init__.py

@@ -0,0 +1 @@
+# Empty file - makes it a package

athena_lab1-0.0.1/src/athena_lab1/core.py

@@ -0,0 +1,24 @@
+import os
+import pandas as pd
+from openai import OpenAI
+from dotenv import load_dotenv
+
+load_dotenv()
+
+KEY = os.environ.get("OPENROUTER_KEY")
+CLIENT = OpenAI(api_key=KEY, base_url="https://openrouter.ai/api/v1")
+
+def load_data(file_path="mydataset.csv"):
+    try:
+        df = pd.read_csv(file_path)
+        return f"\n\nDataset: {df.shape[0]} rows, {df.shape[1]} columns\nColumns: {list(df.columns)}\nFirst 5 rows:\n{df.head()}\n\n"
+    except FileNotFoundError:
+        return ""
+
+def ask(question, file_path="mydataset.csv"):
+    data_text = load_data(file_path)
+    response = CLIENT.chat.completions.create(
+        model="deepseek/deepseek-chat",
+        messages=[{"role": "user", "content": f"{data_text}{question}"}]
+    )
+    print(response.choices[0].message.content)