Implement Kaggle API integration with dataset search, file listing, and download functionality in app.py

.gitignore

@@ -0,0 +1,2 @@
+.env
+.mcp-venv/

app.py

@@ -1,7 +1,58 @@
 import gradio as gr
+import os, json, pathlib
+import pandas as pd
+import tempfile, pathlib
+from typing import List, Dict
+from kaggle.api.kaggle_api_extended import KaggleApi
 
-def greet(name):
-    return "Hello " + name + "!!"
+from dotenv import load_dotenv
+load_dotenv()
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch(mcp_server=True)
+def search_datasets(query: str, max_results: int = 20) -> List[Dict]:
+    """
+    Return brief metadata for up to `max_results` public datasets
+    whose title or description matches `query`.
+    """
+    results = api.dataset_list(search=query, max_size=None, page_size=max_results)
+    out = []
+    for ds in results:
+        out.append({
+            "title": ds.title,
+            "slug": ds.ref,
+            "size_mb": round(ds.size/1e6, 2),
+            "downloads": ds.downloadCount,
+            "votes": ds.totalVotes,
+        })
+    return out
+
+def list_files(dataset_slug: str) -> List[Dict]:
+    files = api.dataset_list_files(dataset_slug).files
+    return [{"name": f.name, "size_mb": round(f.totalBytes / 1e6, 2)} for f in files]
+
+def downlaod_file(dataset_slug: str, file_name: str):
+    tmp_dir = tempfile.mkdtemp()
+    api.dataset_download_file(dataset_slug, file_name, path=tmp_dir, quiet=False)
+    zip_path = pathlib.Path(tmp_dir) / f"{file_name}"
+
+    if not zip_path.exists():
+        zip_path = pathlib.Path(tmp_dir) / f"{file_name}.zip"
+    return str(zip_path)
+
+def _bootstrap_kaggle_credentials():
+    user = os.getenv("KAGGLE_USERNAME")
+    key = os.getenv("KAGGLE_KEY")
+    if not (user and key):
+        raise RuntimeError(
+            "Kaggle credentials not found."
+            "Set KAGGLE_USERNAME and KAGGLE_KEY as env vars or in .env"
+        )
+    cred_path = pathlib.Path.home() / ".kaggle" / "kaggle.json"
+    if not cred_path.exists():
+        cred_path.parent.mkdir(exist_ok=True)
+        cred_path.write_text(json.dumps({"username": user, "key": key}))
+        cred_path.chmod(0o600)
+
+_bootstrap_kaggle_credentials()    
+
+api = KaggleApi()
+api.authenticate()