Update app.py

app.py

@@ -49,17 +49,16 @@ def format_prompt(message, history):
 
 def getProductTitle(history, image):
     product_description=getImageDescription(image)
-    system_prompt="You're a helpful e-commerce marketing assitant working on art products."
     prompt="Our product description is as follows: " + product_description + ". Please write a product title options for it."
     yield interactWithModel(history, system_prompt, prompt)
 
 def getProductDescription(history):
-    system_prompt="You're a helpful e-commerce marketing assitant working on art products."
     prompt="Please also write an SEO friendly description for it describing its value to its users."
     yield interactWithModel(history, system_prompt, prompt)
     
 
-def interactWithModel(history, system_prompt, prompt):
+def interactWithModel(history, prompt):
+    system_prompt="You're a helpful e-commerce marketing assitant working on art products."
     client = InferenceClient("google/gemma-7b-it")
     rand_val = random.randint(1, 1111111111111111)
     if not history:
@@ -185,10 +184,12 @@ with gr.Blocks(css=css) as demo:
     with gr.Row():
         with gr.Column(elem_id="col-container"):
             image = gr.Image(type="pil")
-            chat = gr.Chatbot(show_label=False)
+            output = gr.Image(type="pil", interactive=False)
             submit = gr.Button(value="Upload", variant="primary")
         with gr.Column():
-            output = gr.Image(type="pil", interactive=False)
+            chat = gr.Chatbot(show_label=False)
+            user_input= gr.Textbox()
+            send = gr.Button(value="Send")
         
     title_handler = (
         getProductTitle,
@@ -202,6 +203,12 @@ with gr.Blocks(css=css) as demo:
         [chat]
     )
 
+    interaction_handler = (
+        getProductDescription,
+        [chat, user_input],
+        [chat]
+    )
+
     background_remover_handler = (
         process,
         [image],
@@ -214,8 +221,9 @@ with gr.Blocks(css=css) as demo:
     #     [submit]
     # )
 
-    event = submit.click(*title_handler).then(*description_handler)
-    event2 = submit.click(*background_remover_handler)
+    submit.click(*title_handler).then(*description_handler)
+    submit.click(*background_remover_handler)
+    send.click(*interaction_handler)
     # event.then(*postresponse_handler)
 
 demo.launch()