Update app_dialogue.py

app_dialogue.py

@@ -69,6 +69,12 @@ from PIL import Image
 import tempfile
 
 
+def add_file(file):
+    #history = history + [((file.name,), None)]
+    return file.name
+    #return history
+
+    
 def convert_to_rgb_pil(image):
     """
     Convert a PIL Image object to RGB mode and save it locally.
@@ -553,36 +559,12 @@ textbox = gr.Textbox(
     label="Text input",
 )
 
-css_not = """
-#chatbot {
-background-image: url('https://e7.pngegg.com/pngimages/393/627/png-clipart-obelix-asterix-cartoon-comic-book-comics-asterix-and-obelix-comics-food.png');
-background-repeat: repeat;
-}
-"""
-
-css = """
-#chatbot {
-background-image: url('https://huggingface.co/spaces/ysharma/m4-dialogue_copy4/resolve/main/idefics_720.png');
-background-repeat: repeat;
-}
-"""
-css1 = """
-#chatbot {
-background-image: url('https://huggingface.co/spaces/ysharma/m4-dialogue_copy4/resolve/main/idefics_720_1.png');
-background-repeat: repeat;
-}
-"""
 css2 = """
 #chatbot {
 background-image: url('https://huggingface.co/spaces/ysharma/m4-dialogue_copy4/resolve/main/idefics_720_2.png');
 background-repeat: repeat;}
 """
-css3 = """
-#chatbot {
-background-image: url('https://huggingface.co/spaces/ysharma/m4-dialogue_copy4/resolve/main/idefics_720_3.png');
-background-repeat: repeat;
-}
-"""
+
 
 with gr.Blocks(title="IDEFICS-Chat", theme=gr.themes.Base(), css=css2) as demo:
     gr.Markdown(
@@ -611,7 +593,7 @@ with gr.Blocks(title="IDEFICS-Chat", theme=gr.themes.Base(), css=css2) as demo:
                 )
             processor, tokenizer, model = load_processor_tokenizer_model(model_selector.value)
 
-            imagebox = gr.Image(type="filepath", label="Image input")
+            imagebox = gr.Image(type="filepath", label="Image input", visible=False)
 
             with gr.Accordion("Advanced parameters", open=False, visible=True) as parameter_row:
                 max_new_tokens = gr.Slider(
@@ -802,17 +784,22 @@ And so, the story of Mulan and Shrek's romance came to an end, leaving a lasting
                     ],
                 ],
             )
-            with gr.Row():
-                with gr.Column(scale=8):
-                    textbox.render()
-                with gr.Column(scale=1, min_width=60):
-                    submit_btn = gr.Button(value="Submit", visible=True)
-                with gr.Column(scale=1, min_width=20):
-                    clear_btn = gr.ClearButton([textbox, chatbot])
-                with gr.Column(scale=1, min_width=15):
-                    dope_bttn = gr.Button("Dope🔥")
-                with gr.Column(scale=1, min_width=15):
-                    problematic_bttn = gr.Button("Problematic😬")
+            with gr.Group():
+                with gr.Row():
+                    with gr.Column(scale=7):
+                        textbox.render()
+                    with gr.Column(scale=1, min_width=80):
+                        submit_btn = gr.Button(value="Submit", visible=True)
+                    with gr.Column(scale=1, min_width=10):
+                        clear_btn = gr.ClearButton([textbox, chatbot])
+                    with gr.Column(scale=1, min_width=10):
+                        upload_btn = gr.UploadButton("📁", file_types=["image"])
+            with gr.Group():
+                with gr.Row():
+                    with gr.Column(scale=1, min_width=50):
+                        dope_bttn = gr.Button("Dope🔥")
+                    with gr.Column(scale=1, min_width=50):
+                        problematic_bttn = gr.Button("Problematic😬")
 
     def model_inference(
         user_prompt_str,
@@ -926,6 +913,8 @@ And so, the story of Mulan and Shrek's romance came to an end, leaving a lasting
         ],
     )
 
+    upload_btn.upload(add_file, [btn], [imagebox], queue=False) 
+
     # Using Flagging for saving dope and problematic examples
     # Dope examples flagging
     dope_hf_callback.setup(