Update app.py

app.py

@@ -28,6 +28,26 @@ import torch
 import mysql.connector
 from mysql.connector import errorcode, pooling
 from dotenv import load_dotenv
+from huggingface_hub import login
+
+# Initialize a zero-shot classification pipeline
+classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
+
+def parse_command(message):
+    candidate_labels = ["filter", "sort", "export", "log"]
+    result = classifier(message, candidate_labels)
+    return result['labels'][0] if result['scores'][0] > 0.5 else None
+
+# Usage
+command = parse_command("Filter apples, oranges in column Description")
+print(command)  # Output: 'filter'
+
+HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
+if not HUGGINGFACE_TOKEN:
+    raise ValueError("HUGGINGFACE_TOKEN is not set in the environment variables.")
+
+login(token=HUGGINGFACE_TOKEN)
+
 
 # Load environment variables from .env file
 load_dotenv()