Update app.py

app.py

@@ -47,9 +47,9 @@ def format_prompt(message, history):
         prompt += f"<start_of_turn>user{message}<end_of_turn><start_of_turn>modelo"
         return prompt
 
-def getProductTitle(history, image):
+def getProductTitle(history, context, image):
     product_description=getImageDescription(image)
-    prompt="Our product description is as follows: " + product_description + ". Please write a product title options for it."
+    prompt="We have a product which is a" + context + ". Product description is as follows: " + product_description + ". Please write a product title options for it."
     yield interactWithModel(history, prompt)
 
 def getProductDescription(history):
@@ -185,6 +185,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Column(elem_id="col-container"):
             image = gr.Image(type="pil")
             output = gr.Image(type="pil", interactive=False)
+            context = gr.Textbox(label="Small description")
             submit = gr.Button(value="Upload", variant="primary")
         with gr.Column():
             chat = gr.Chatbot(show_label=False)
@@ -193,7 +194,7 @@ with gr.Blocks(css=css) as demo:
         
     title_handler = (
         getProductTitle,
-        [chat, image],
+        [chat, context, image],
         [chat]
     )