Food_Classifier / nutrition_fetcher.py
nabeel857's picture
Upload 6 files
dd5ae0f verified
raw
history blame contribute delete
626 Bytes
import requests
import pandas as pd
# Function to fetch nutrition details
def fetch_nutrition(query: str):
api_url = f'https://api.api-ninjas.com/v1/nutrition?query={query}'
headers = {'X-Api-Key': 'Hu4DkNyaIFT+E/FfCPRaYw==EUSFTIrXpCdQXrjH'}
response = requests.get(api_url, headers=headers)
if response.status_code == requests.codes.ok:
data = response.json()
if data:
df = pd.DataFrame(data)
# Drop unwanted columns
df = df.drop(columns=['calories', 'serving_size_g', 'protein_g'], errors='ignore')
return df
return None