Update app.py

app.py

@@ -425,7 +425,8 @@ def invoke (prompt, file, history, rag_option, model_option, openai_api_key, k=3
         else:
             print("LLM aufrufen ohne RAG: ...........")
             if (prompt.find('zeichnen') != -1):
-                result = response.content
+                #result = response.content 
+                result = response.json()["images"]
             else:
                 print("ganz normale Ausführung.........................")
                 result = llm_chain(llm, history_text_und_prompt)
@@ -437,8 +438,11 @@ 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)) 
-        history[-1][1] = image #file.name, 
+        #image = Image.open(io.BytesIO(result)) 
+        for image in result:
+            image_b64 = (f"data:image/jpeg;base64,{image}")
+            images.append(image_b64)
+        history[-1][1] = images[0]  #file.name, 
         print("history zeichnen......................")
         print(history)
         if shared_state.interrupted: