Update app.py

app.py

@@ -357,6 +357,10 @@ def create_picture(history,  prompt):
 # using the Image module from PIL library to view the image 
 #Image.open(response.raw)
 
+def umwandeln_fuer_anzeige(image):
+        buffer = io.BytesIO()  
+        image.save(buffer, format='PNG')
+        return buffer.getvalue()
   
 def invoke (prompt, file, 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,):
     global splittet
@@ -439,13 +443,8 @@ def invoke (prompt, file, history, rag_option, model_option, openai_api_key, k=3
     if (prompt.find('zeichnen') != -1):
         #Bild ausgeben
         image = Image.open(io.BytesIO(result)) 
-        #buffer = io.BytesIO()
-        #image.save(buffer, format='PNG')
-        #desiredObject = buffer.getbuffer()
-        b = io.BytesIO()  
-        image.save(b, format='PNG')
         
-        history[-1][1] =  "<img src='data:image/png;base64,{0}'/>".format(b64encode(b.getvalue()).decode('utf-8'))  #file.name, 
+        history[-1][1] =  "<img src='data:image/png;base64,{0}'/>".format(b64encode(umwandeln_fuer_anzeige(image)).decode('utf-8'))   
         print("history zeichnen......................")
         print(history)
         if shared_state.interrupted: