Update app.py

app.py

@@ -150,6 +150,15 @@ def add_text(chatbot, history, prompt, file):
     print("chatbot nach add_text............")
     print(chatbot)
     return chatbot, history, prompt, "" #gr.Image( label=None, size=(30,30), visible=False, scale=1) #gr.Textbox(value="", interactive=False)
+
+    def add_text2(chatbot, prompt):
+        if (prompt == ""):
+            chatbot = chatbot + [("", "Prompt fehlt!")]
+        else:
+            chatbot = chatbot + [(prompt, None)]
+        print("chatbot nach add_text............")
+        print(chatbot)
+        return chatbot, prompt,
     
 ############################################
 #nach dem Upload soll das zusätzliche Fenster mit dem image drinnen angezeigt werden
@@ -414,22 +423,21 @@ def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, o
             
         chatbot[-1][1] = result
          
-        return chatbot, history, "Success" 
-        """ 
+ 
         for character in result:
-            history[-1][1] += character
+            chatbot[-1][1] += character
             time.sleep(0.03)
-            yield history, "Generating"
+            yield chatbot, history "Generating"
             if shared_state.interrupted:
                 shared_state.recover()
                 try:
-                    yield history, "Stop: Success"
+                    yield chatbot, history "Stop: Success"
                 except:
                     pass
-        """
+        
 ##################################################
 #zu einem Text-Prompt ein Bild via Stable Diffusion generieren
-def generate_bild(prompt, chatbot, temperature=0.5, max_new_tokens=4048,top_p=0.6, repetition_penalty=1.3,):
+def generate_bild(prompt, chatbot,  temperature=0.5, max_new_tokens=4048,top_p=0.6, repetition_penalty=1.3):
     #Bild nach Anweisung zeichnen und in History darstellen...
     data = {"inputs": prompt}
     response = requests.post(API_URL, headers=HEADERS, json=data)
@@ -439,7 +447,7 @@ def generate_bild(prompt, chatbot, temperature=0.5, max_new_tokens=4048,top_p=0.
     image = Image.open(io.BytesIO(result)) 
     image_64 = umwandeln_fuer_anzeige(image)
     chatbot.append(prompt, "<img src='data:image/png;base64,{0}'/>".format(base64.b64encode(image_64).decode('utf-8')))
-    return chatbot
+    return chatbot, "Success"
 
     
 
@@ -622,32 +630,25 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
     ################################################
     # Tab zum Zeichnen mit Stable Diffusion
     ################################################
-    with gr.Tab("KI zum Zeichnen"):
-        with gr.Row():
-            gr.HTML("LI Zeichnen mit KI")
-        gr.Markdown(description_top)
-        with gr.Row():
-            description2 = "<strong>Information:</strong> Hier wird ein <strong>Large Language Model (LLM)</strong> zum Zeichnen verwendet. Zur Zeit wird hier Stable Diffusion verwendet.\n\n"
-            additional_inputs = [
+    description2 = "<strong>Information:</strong> Hier wird ein <strong>Large Language Model (LLM)</strong> zum Zeichnen verwendet. Zur Zeit wird hier Stable Diffusion verwendet.\n\n"
+    additional_inputs = [
                 gr.Slider(label="Temperature", value=0.65, minimum=0.0, maximum=1.0, step=0.05, interactive=True, info="Höhere Werte erzeugen diversere Antworten", visible=True),
                 gr.Slider(label="Max new tokens", value=1024, minimum=0, maximum=4096, step=64, interactive=True, info="Maximale Anzahl neuer Tokens", visible=True),
                 gr.Slider(label="Top-p (nucleus sampling)", value=0.6, minimum=0.0, maximum=1, step=0.05, interactive=True, info="Höhere Werte verwenden auch Tokens mit niedrigerer Wahrscheinlichkeit.", visible=True),
                 gr.Slider(label="Repetition penalty", value=1.2, minimum=1.0, maximum=2.0, step=0.05, interactive=True, info="Strafe für wiederholte Tokens", visible=True)
             ]
+    with gr.Tab("KI zum Zeichnen"):
+        with gr.Row():
+            gr.HTML("LI Zeichnen mit KI")
+            status_display2 = gr.Markdown("Success", elem_id="status_display")
+        gr.Markdown(description2)
+        with gr.Row():
+                        
             chatbot_bild = gr.Chatbot(elem_id="li-zeichnen")
-            chat_interface_bild = gr.ChatInterface(fn=generate_bild, 
-                additional_inputs = additional_inputs,
-                additional_inputs_accordion = gr.Accordion(label="Weitere Eingaben...",  open=False),
-                title = "Zeichnen mit KI",
-                theme="soft",
-                chatbot=chatbot_bild,
-                retry_btn="🔄 Wiederholen",
-                undo_btn="↩️ Letztes löschen",
-                clear_btn="🗑️ Verlauf löschen",
-                submit_btn = "Abschicken",
-                description = description2) 
-            
-    gr.Markdown(description)
+            submitBtn2 = gr.Button("Senden")
+            cancelBtn2 = gr.Button("Stop")
+            emptyBtn2 = gr.ClearButton([user_input, chatbot, history, file_display, image_display], value="🧹 Neue Session", scale=10)  
+            additional_inputs_accordion = gr.Accordion(label="Weitere Eingaben...",  open=False)
     
     ######################################
     # Events und Übergabe Werte an Funktionen 
@@ -660,7 +661,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
         inputs=[
             user_question,
             upload,
-            chatbot,
+            chatbot_bild,
             history,
             rag_option,
             model_option,
@@ -683,7 +684,7 @@ 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], show_progress=True
+        fn=add_text, inputs=[chatbot, history, user_input, upload], outputs=[chatbot, user_question, user_input], show_progress=True
     )
 
     predict_event1 = user_input.submit(**transfer_input_args, queue=False,).then(**predict_args)
@@ -697,6 +698,26 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
 
     ######################################
     # Für Tab 2: Zeichnen
+    predict_args2 = dict(
+        fn=generate_bild,
+        inputs=[
+            user_question2,
+            chatbot_bild,
+            additional_inputs,
+        ],
+        outputs=[chatbot_bild, status_display2], #[chatbot, history,  status_display]
+        show_progress=True,
+    )
+    transfer_input_args = dict(
+        fn=add_text2, inputs=[chatbot_bild, user_input2], outputs=[chatbot_bild, user_question2, user_input2], show_progress=True
+    )
+    predict_event2_1 = user_input.submit(**transfer_input_args2, queue=False,).then(**predict_args2)
+    predict_event2_2 = submitBtn.click(**transfer_input_args2, queue=False,).then(**predict_args2)
+    #emptyBtn2.click(clear_all, [], [file_display, image_display])
+    
+    cancelBtn.click(
+        cancels=[predict_event2_1,predict_event2_2 ]
+    )
     
     
 demo.title = "LI-ChatBot"