Update app.py

app.py

@@ -136,7 +136,7 @@ os.environ["HUGGINGFACEHUB_API_TOKEN"] = HUGGINGFACEHUB_API_TOKEN
 ##############################################
 #wenn löschen Button geklickt
 def clear_all():
-    return gr.File(visible=False)
+    return gr.File(visible=False), gr.Image(visible=False)
 
 ##############################################
 #History - die Frage oder das File eintragen...
@@ -150,12 +150,12 @@ def add_text(chatbot, history, prompt, file):
             chatbot = chatbot + [((file.name,), None), (prompt, None)]
     print("chatbot nach add_text............")
     print(chatbot)
-    return chatbot, history, prompt, "", gr.File( label=None, interactive=False,  height=20, min_width=20, visible=False, scale=2) #gr.Textbox(value="", interactive=False)
+    return chatbot, history, prompt, "" #gr.Image( label=None, size=(30,30), visible=False, scale=1) #gr.Textbox(value="", interactive=False)
     
 ############################################
 #nach dem Upload soll das zusätzliche Fenster mit dem image drinnen angezeigt werden
 def file_anzeigen(file):
-    return gr.File(visible=True), file
+    return gr.Image(visible=True), file
 
 ##########################################
 #Hilfsfunktion, um ein von Stable Diffusion erzeugtes Bild für die Ausgabe in der History vorzubereiten
@@ -411,7 +411,7 @@ def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, o
         image_64 = umwandeln_fuer_anzeige(image)
         chatbot[-1][1] =  "<img src='data:image/png;base64,{0}'/>".format(base64.b64encode(image_64).decode('utf-8')) 
         history = history + [(prompt, result)]  
-        return chatbot, history, gr.File(visible=False), "Success"
+        return chatbot, history, "Success"
     else:
         #kein Bild hochgeladen -> auf Text antworten...
         if (file == None):
@@ -426,7 +426,7 @@ def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, o
             
         chatbot[-1][1] = result
          
-        return chatbot, history, gr.File(visible=False), "Success" 
+        return chatbot, history, "Success" 
         """ 
         for character in result:
             history[-1][1] += character
@@ -572,9 +572,10 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                 with gr.Column(min_width=70, scale=1):
                     cancelBtn = gr.Button("Stop")
             with gr.Row():
-                file_display = gr.File( label=None, interactive=False,  height=20, min_width=20, visible=False, scale=2)
+                file_display = gr.File( label=None, interactive=False,  height=20, min_width=20, visible=False, scale=0)
+                image_display = gr.Image(label = None, shape=[30,30], visible=False, scale = 1)
                 upload = gr.UploadButton("📁", file_types=["image", "video", "audio"], scale=3)
-                emptyBtn = gr.ClearButton([user_input, chatbot, history, file_display], [file_display], value="🧹 Neue Session", scale=3)  
+                emptyBtn = gr.ClearButton([user_input, chatbot, history, file_display, image_display], value="🧹 Neue Session", scale=3)  
 
         with gr.Column():
             with gr.Column(min_width=50, scale=1):
@@ -638,7 +639,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
             max_context_length_tokens,
             repetition_penalty
         ],
-        outputs=[chatbot, history, file_display, status_display], #[chatbot, history,  status_display]
+        outputs=[chatbot, history, status_display], #[chatbot, history,  status_display]
         show_progress=True,
     )
 
@@ -649,13 +650,13 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
 
     # Chatbot
     transfer_input_args = dict(
-        fn=add_text, inputs=[chatbot, history, user_input, upload], outputs=[chatbot, history, user_question, user_input, file_display], show_progress=True
+        fn=add_text, inputs=[chatbot, history, user_input, upload], outputs=[chatbot, history, user_question, user_input], show_progress=True
     )
 
     predict_event1 = user_input.submit(**transfer_input_args, queue=False,).then(**predict_args)
     predict_event2 = submitBtn.click(**transfer_input_args, queue=False,).then(**predict_args)
-    predict_event3 = upload.upload(file_anzeigen, [upload], [file_display, file_display] ) #.then(**predict_args)
-    emptyBtn.click(clear_all, [], [file_display])
+    predict_event3 = upload.upload(file_anzeigen, [upload], [image_display, file_display] ) #.then(**predict_args)
+    emptyBtn.click(clear_all, [], [file_display, image_display])
     
     cancelBtn.click(
         cancels=[predict_event1,predict_event2, predict_event3 ]