main.py

from taipy.gui import Gui, notify
from vision.HTNImageDetection import imageDetection
from textGenerator import textGenerator

name = ""
pref = ""  
image = ""
size = 0
desc = ""

items = {}

def getMeals(name, pref, size, image):  
    if image:
        meals = str(textGenerator([name, size, imageDetection(image), pref]))
    # TODO: Object Classification using YOLO Code goes here
    # meals=[f'Hello {name}, here is few {pref} meals you can make for {size} people:','1. Contrary to popular belief:', 'Lorem Ipsum is not simply random text It has roots in a piece of classical Latin literature from 45 BC', '2. Contrary to popular belief:', 'Lorem Ipsum is not simply random text It has roots in a piece of classical Latin literature from 45 BC', '3. Contrary to popular belief:', 'Lorem Ipsum is not simply random text It has roots in a piece of classical Latin literature from 45 BC'] 
    else:
        meals = "No image provided!"
    return meals

# class User:
#     def getItems(image):
#         items = {'apple':1}  # 'item: # of items'
#         # TODO: Object Classification using YOLO Code goes here
#         return items
#     def __init__(self, name, size, image):
#         self.name = name
#         self.size = size
#         self.items = getItems(image)

#          # TODO: Research and create prompt for ideal results
#         prompt = open("prompt.txt", "r").read()
#         # TODO: Plug in items available and their quantities
#         # TODO: Send this prompt to BentoML
#         ideas = "Hello" + name
#         # ideas = BentoML(...)
#         return ideas

page = """
Name:  \n
<|{name}|input|>


Portion Size: <|{size}|>   
<|{size}|slider|min=1|max=10|>


Preferences:  
<|{pref}|input|>


Input a image of your fridge: <|{image}|>   
<|{image}|file_selector|extensions=.png,.jpg,.jpeg,.heic|>


<|Submit|button|on_action=submit|>


<|{desc}|>

"""

def submit(state):
    state.desc = getMeals(state.name, state.pref, state.size, state.image)
    # state.d, state.h1, state.d1, state.h2, state.d2, state.h3, state.d3 = getMeals(state.name, state.pref, state.size, state.image)

if __name__ == "__main__":
    gui = Gui(page=page)
    gui.run(title="Fridge Guru")


# <|{h1}|>

# <|{d1}|>


# <|{h2}|>

# <|{d2}|>


# <|{h3}|>

# <|{d3}|>