Spaces:
Running
Running
| import os | |
| from dotenv import load_dotenv | |
| from supabase import create_client | |
| import requests | |
| # Load environment variables from .env file | |
| load_dotenv() | |
| # Supabase API Config | |
| SUPABASE_URL = "https://lmpazoxzucnlqqxjoihi.supabase.co" | |
| SUPABASE_KEY = os.getenv("SUPABASE_API_KEY") | |
| if not SUPABASE_KEY: | |
| raise ValueError("SUPABASE_KEY is not set in the environment variables.") | |
| supabase = create_client(SUPABASE_URL, SUPABASE_KEY) | |
| HF_MODELS = { | |
| "gemma": "https://api-inference.huggingface.co/models/google/gemma-7b", | |
| "bart": "https://api-inference.huggingface.co/models/facebook/bart-large-cnn" | |
| } | |
| # Hugging Face API Config | |
| #HF_API_URL = "https://router.huggingface.co/hf-inference/models/google/gemma-7b" | |
| HF_API_TOKEN = os.getenv("HF_API_TOKEN") | |
| HF_HEADERS = {"Authorization": f"Bearer HF_API_TOKEN"} | |
| # Ensure the API key is loaded | |
| if not HF_API_TOKEN: | |
| raise ValueError("Missing Hugging Face API key. Check your .env file.") | |
| # | |
| def query(payload, model="gemma"): | |
| """ | |
| Sends a request to the selected Hugging Face model API. | |
| :param payload: The input data for inference. | |
| :param model: Choose either 'gemma' (for google/gemma-7b) or 'bart' (for facebook/bart-large-cnn). | |
| :return: The model's response in JSON format, or None if the request fails. | |
| """ | |
| if model not in HF_MODELS: | |
| raise ValueError("Invalid model name. Choose 'gemma' or 'bart'.") | |
| api_url = f"https://api-inference.huggingface.co/models/{HF_MODELS[model]}" | |
| try: | |
| response = requests.post(api_url, headers=HF_HEADERS, json=payload) | |
| if response.status_code == 401: | |
| print(f"Error querying Hugging Face model '{model}': 401 Unauthorized. Check API key.") | |
| return None # Handle authentication failure | |
| response.raise_for_status() # Raise an error for failed requests (e.g., 500 errors) | |
| return response.json() # Return the parsed JSON response | |
| except requests.exceptions.RequestException as e: | |
| print(f"Error querying Hugging Face model '{model}': {e}") | |
| return None # Return None if API call fails | |
| # Bart query | |
| def query(payload, model="bart"): | |
| """ | |
| Sends a request to the selected Hugging Face model API. | |
| :param payload: The input data for inference. | |
| :param model: Choose either 'gemma' (for google/gemma-7b) or 'bart' (for facebook/bart-large-cnn). | |
| :return: The model's response in JSON format, or None if the request fails. | |
| """ | |
| if model not in HF_MODELS: | |
| raise ValueError("Invalid model name. Choose 'gemma' or 'bart'.") | |
| api_url = f"https://api-inference.huggingface.co/models/{HF_MODELS[model]}" | |
| try: | |
| response = requests.post(api_url, headers=HF_HEADERS, json=payload) | |
| if response.status_code == 401: | |
| print(f"Error querying Hugging Face model '{model}': 401 Unauthorized. Check API key.") | |
| return None # Handle authentication failure | |
| response.raise_for_status() # Raise an error for failed requests (e.g., 500 errors) | |
| return response.json() # Return the parsed JSON response | |
| except requests.exceptions.RequestException as e: | |
| print(f"Error querying Hugging Face model '{model}': {e}") | |
| return None # Return None if API call fails |