from io import BytesIO
import pandas as pd
import numpy as np
import requests
import streamlit as st
import easyocr
import PIL
from PIL import Image, ImageDraw
from matplotlib import pyplot as plt
import streamlit.components.v1 as components
import re
from streamlit import session_state
# from streamlit_lottie import st_lottie
import streamlit as st
import os
from dotenv import load_dotenv, find_dotenv
st.set_page_config(
page_title="Intro",
layout='wide'
)
# Load environment variables
_ = load_dotenv(find_dotenv())
# Get OpenAI API key
openai_api_key = os.getenv('OPENAI_API_KEY')
# Google Drive file ID for the video
video_file_id = '1miZ6Zz4lt0wxTGbrtR8IltVr_fOPX6Vr'
video_url = f'https://drive.google.com/uc?export=view&id={video_file_id}'
@st.cache_data
def initialize_grocery_data():
return pd.DataFrame(columns=['Name', 'Quantity', 'Additional Notes/Expiration Date'])
def manually_input_items(grocery_data):
return grocery_data
def upload_receipt():
st.header("Upload Receipt")
uploaded_file = st.file_uploader("Choose a receipt file", type=["jpg", "jpeg", "png", "pdf"])
if uploaded_file is not None:
st.success("Receipt uploaded successfully!")
# Add your code here for processing the uploaded receipt
if 'clicked' not in st.session_state:
st.session_state.clicked = False
if 'total_groceries' not in st.session_state:
st.session_state.total_groceries = {}
if 'grocery_data' not in st.session_state:
st.session_state.grocery_data = pd.DataFrame(columns=['Name', 'Quantity', 'Additional Notes/Expiration Date'])
if 'editor_key' not in st.session_state:
st.session_state.editor_key = 0
if 'manual_entry' not in st.session_state:
st.session_state.manual_entry = True
if 'file_uploaded' not in st.session_state:
st.session_state.file_uploaded = False
if 'ocr_extracted_values' not in st.session_state:
session_state.ocr_extracted_values = pd.DataFrame()
def click_save():
session_state.clicked = True
def switch_style():
session_state.manual_entry = not session_state.manual_entry
# Define the HTML and CSS styling for the title
title_html = """
Nom Nom!
"""
# Use st.markdown to display the title with the specified font
st.markdown(title_html, unsafe_allow_html=True)
caption_html = """
Helping College Students Keep Track of Groceries
"""
st.markdown(caption_html, unsafe_allow_html=True)
ingredient_quantities = {}
# Initialize or retrieve the existing dataframe to store grocery items
#grocery_data = initialize_grocery_data()
tab1, tab2 = st.tabs(["Add Groceries", "Available Groceries"])
df = pd.DataFrame()
with tab1:
st.button("Switch to OCR Input", on_click=switch_style)
col1, col2 = st.columns([1.5, 1])
if session_state.manual_entry:
with col1:
if not session_state.clicked:
st.subheader("Add Groceries - Manually Input Items")
session_state.grocery_data = st.data_editor(pd.DataFrame(columns=['Name', 'Quantity', 'Additional Notes/Expiration Date']), key=session_state.editor_key, num_rows="dynamic", hide_index=True, use_container_width=True)
print(session_state.grocery_data)
st.button("Press me to update Tab 2", on_click=click_save)
else:
session_state.editor_key += 1
with col2:
file_id = '1_3FjYmWlJcY182qnLIjFPTZYj4gneUMg'
base_url = 'https://drive.google.com/uc?id='
image_url = base_url + file_id
response = requests.get(image_url)
img = Image.open(BytesIO(response.content))
# image1 = Image.open(r'https://drive.google.com/file/d/1_3FjYmWlJcY182qnLIjFPTZYj4gneUMg/view?usp=sharing')
st.image(img, width=400)
st.write("---")
else:
with col1:
st.subheader("Upload receipt instead")
uploaded_file = st.file_uploader("Choose a receipt file", type=["jpg", "jpeg", "png"])
if uploaded_file is not None:
session_state.file_uploaded = True
st.success("Receipt uploaded successfully!")
st.write("")
if session_state.file_uploaded:
receipt_image = Image.open(uploaded_file)
st.image(receipt_image)
reader = easyocr.Reader(['en'], gpu=False)
result = reader.readtext(np.array(receipt_image))
text_list = []
confidence_list = []
for idx in range(len(result)):
pred_coor = result[idx][0]
pred_text = result[idx][1]
pred_confidence = result[idx][2]
text_list.append(pred_text)
confidence_list.append(pred_confidence)
df = pd.DataFrame({'Text': text_list})
with col2:
if session_state.file_uploaded:
st.subheader("Updated Grocery List")
df = df[~df['Text'].str.contains(r'[\d]')]
df['Add to List?'] = None
df2 = st.data_editor(df, use_container_width=True, num_rows = 'dynamic', column_config={"Text": "Item",
"Add to List?" : st.column_config.CheckboxColumn(
"Add to List?",
help="Select if you want to add this to the list.",
default=False
)}, key="OCR_DataFrame")
session_state.ocr_extracted_values = df2
st.button("Press me to update Tab 2", on_click=click_save)
def ingredient_html_maker(ingredient, num_ingredients, map_for_grocery):
return_val = f"