Update app.py

app.py

@@ -132,16 +132,16 @@ os.environ["HUGGINGFACEHUB_API_TOKEN"] = HUGGINGFACEHUB_API_TOKEN
 
 ##############################################
 #History - die Frage oder das File eintragen...
-def add_text(history, prompt, file):
+def add_text(chatbot, history, prompt, file):
     if (file == None):
-        history = history + [(prompt, None)]
+        chatbot = chatbot + [(prompt, None)]
     else:
         if (prompt == ""):
-            history = history + [((file.name,), "Prompt fehlt!")]
+            chatbot = chatbot + [((file.name,), "Prompt fehlt!")]
         else:
-            history = history + [((file.name,), None), (prompt, None)]
+            chatbot = chatbot + [((file.name,), None), (prompt, None)]
             
-    return history, 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.File( label=None, interactive=False,  height=20, min_width=20, visible=False, scale=2) #gr.Textbox(value="", interactive=False)
     
 def add_file(history, file, prompt):
     if (prompt == ""):
@@ -390,17 +390,19 @@ def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, o
         #Bild ausgeben
         image = Image.open(io.BytesIO(result)) 
         image_64 = umwandeln_fuer_anzeige(image)
-        history[-1][1] =  "<img src='data:image/png;base64,{0}'/>".format(b64encode(image_64).decode('utf-8'))   
+        chatbot[-1][1] =  "<img src='data:image/png;base64,{0}'/>".format(b64encode(image_64).decode('utf-8')) 
+        history[-1][1] = b64encode(image_64).decode('utf-8')
         print("history zeichnen......................")
-        print(history)
-        return history, "Success"
+        print(chatbot)
+        return chatbot, history, "Success"
      else:
         result = generate_text(prompt, file, chatbot, history, rag_option, model_option, openai_api_key, k=3, top_p=0.6, temperature=0.5, max_new_tokens=4048, max_context_length_tokens=2048, repetition_penalty=1.3,)
         #Antwort als Stream ausgeben... wenn Textantwort gefordert
+        chatbot[-1][1] = result
         history[-1][1] = result
         print("history nach Zusatz und mit KI Antwort...........")
-        print(history)
-        return history, history, "Success" 
+        print(chatbot)
+        return chatbot, history, "Success" 
         """ 
         for character in result:
             history[-1][1] += character
@@ -438,6 +440,7 @@ def generate_text (prompt, file, chatbot, history, rag_option, model_option, ope
         history_text_und_prompt = generate_prompt_with_history_openai(prompt, history)
     else:
         prompt_neu = prompt + file.path, #b64encode(file).decode("utf-8")
+        print("prompt_neu............................")
         print(prompt_neu)
         history_text_und_prompt = generate_prompt_with_history_openai(prompt_neu, history)
         
@@ -612,7 +615,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
 
     # Chatbot
     transfer_input_args = dict(
-        fn=add_text, inputs=[chatbot, 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, file_display], show_progress=True
     )
 
     predict_event1 = user_input.submit(**transfer_input_args, queue=False,).then(**predict_args)