Update app.py

app.py

@@ -177,8 +177,6 @@ def cancel_outputing():
 def reset_textbox():
     return gr.update(value=""),""
 
-def reset_file():
-    return gr.Image(visible = False), None 
     
 ##########################################
 #Hilfsfunktion, um ein von Stable Diffusion erzeugtes Bild für die Ausgabe in der History vorzubereiten
@@ -479,11 +477,11 @@ def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, o
         for character in result:
             chatbot[-1][1] += character
             time.sleep(0.03)
-            yield chatbot, history, "Generating"
+            yield chatbot, history, None, "Generating"
             if shared_state.interrupted:
                 shared_state.recover()
                 try:
-                    yield chatbot, history, "Stop: Success"
+                    yield chatbot, history, None, "Stop: Success"
                 except:
                     pass
         
@@ -767,13 +765,9 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
             max_context_length_tokens,
             repetition_penalty
         ],
-        outputs=[chatbot, history, status_display], #[chatbot, history,  status_display]
+        outputs=[chatbot, history, attached_file, status_display], #[chatbot, history,  status_display]
         show_progress=True,
     )
-
-    end_args = dict(
-        fn=reset_file, inputs=[], outputs=[image_display, attached_file]
-    )
     
     reset_args = dict(
         fn=reset_textbox, inputs=[], outputs=[user_input, status_display]
@@ -784,8 +778,8 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
         fn=add_text, inputs=[chatbot, history, user_input, attached_file], outputs=[chatbot, history, user_question, attached_file, image_display , user_input], show_progress=True
     )
 
-    predict_event1 = user_input.submit(**transfer_input_args, queue=False,).then(**predict_args).then(**end_args)
-    predict_event2 = submitBtn.click(**transfer_input_args, queue=False,).then(**predict_args).then(**end_args)
+    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], [image_display, image_display, attached_file] ) #.then(**predict_args)
     emptyBtn.click(clear_all, [], [attached_file, image_display])
     image_display.select(file_loeschen, [], [attached_file, image_display])