Update app.py

app.py

@@ -386,27 +386,30 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, rag_option
             
 ##################################################
 #zu einem Text-Prompt ein Bild via Stable Diffusion generieren
-def generate_bild(prompt, chatbot, model_option_zeichnen='HuggingFace', temperature=0.5, max_new_tokens=4048,top_p=0.6, repetition_penalty=1.3):
+def generate_bild(prompt, chatbot, model_option_zeichnen='HuggingFace', temperature=0.5, max_new_tokens=4048,top_p=0.6, repetition_penalty=1.3, validate=False):
     global client
-    if (model_option_zeichnen == "Stable Diffusion"):
-        print("Bild Erzeugung HF..............................")
-        #Bild nach Anweisung zeichnen und in History darstellen...
-        data = {"inputs": prompt}
-        response = requests.post(API_URL, headers=HEADERS, json=data)
-        print("fertig Bild")
-        result = response.content 
-        #Bild ausgeben
-        image = Image.open(io.BytesIO(result)) 
-        image_64 = umwandeln_fuer_anzeige(image)
-        chatbot[-1][1]= "<img src='data:image/png;base64,{0}'/>".format(base64.b64encode(image_64).decode('utf-8'))
-    else:     
-        print("Bild Erzeugung DallE..............................")
-        #als Format ginge auch 'url', n - Anz. der erzeugten Bilder
-        response = client.images.generate(model="dall-e-3",prompt=prompt,size="1024x1024",quality="standard",n=1, response_format='b64_json') 
-        #chatbot[-1][1]= "<img src='data:image/png;base64,{0}'/>".format(base64.b64encode(image_64).decode('utf-8'))
-        chatbot[-1][1] = "<img src='data:image/png;base64,{0}'/>".format(response.data[0].b64_json)
-    
-    return chatbot, "Antwort KI: Success"
+    if (validate):
+        if (model_option_zeichnen == "Stable Diffusion"):
+            print("Bild Erzeugung HF..............................")
+            #Bild nach Anweisung zeichnen und in History darstellen...
+            data = {"inputs": prompt}
+            response = requests.post(API_URL, headers=HEADERS, json=data)
+            print("fertig Bild")
+            result = response.content 
+            #Bild ausgeben
+            image = Image.open(io.BytesIO(result)) 
+            image_64 = umwandeln_fuer_anzeige(image)
+            chatbot[-1][1]= "<img src='data:image/png;base64,{0}'/>".format(base64.b64encode(image_64).decode('utf-8'))
+        else:     
+            print("Bild Erzeugung DallE..............................")
+            #als Format ginge auch 'url', n - Anz. der erzeugten Bilder
+            response = client.images.generate(model="dall-e-3",prompt=prompt,size="1024x1024",quality="standard",n=1, response_format='b64_json') 
+            #chatbot[-1][1]= "<img src='data:image/png;base64,{0}'/>".format(base64.b64encode(image_64).decode('utf-8'))
+            chatbot[-1][1] = "<img src='data:image/png;base64,{0}'/>".format(response.data[0].b64_json)
+        
+        return chatbot, "Antwort KI: Success"
+    else: #noch nicht validiert ...
+        return chatbot, "Bitte erst validieren!"
     
 
 ##################################################