Update app.py

app.py

@@ -393,6 +393,11 @@ def invoke (prompt, file, history, rag_option, model_option, openai_api_key, k=3
                 #print("OpenAI zeichnen.......................")
                 #llm = ChatOpenAI(model_name = MODEL_NAME_OAI_ZEICHNEN,  openai_api_key = openai_api_key, temperature=temperature)#, top_p = top_p)  
                 data = {"inputs": prompt}
+                response = requests.post(API_URL, headers=HEADERS, json=data)
+                result = response.content
+                image = Image.open(io.BytesIO(result))
+                history[-1][1] = image
+                return history, "Stop: Success"
             else: 
                 print("OpenAI normal.......................")
                 llm = ChatOpenAI(model_name = MODEL_NAME,  openai_api_key = openai_api_key, temperature=temperature)#, top_p = top_p)
@@ -421,35 +426,26 @@ def invoke (prompt, file, history, rag_option, model_option, openai_api_key, k=3
             result = rag_chain(llm, history_text_und_prompt, db)
         else:
             print("LLM aufrufen ohne RAG: ...........")
-            if (prompt.find('Bild zeichnen') != -1):
-                response = requests.post(API_URL, headers=HEADERS, json=data)
-                result = response.content
-                image = Image.open(io.BytesIO(result))
-            else:
-                result = llm_chain(llm, history_text_und_prompt)
+            result = llm_chain(llm, history_text_und_prompt)
             
 
     except Exception as e:
         raise gr.Error(e)
 
 
-    if (prompt.find('Bild zeichnen') != -1):
-        history[-1][1] = image
-        return history, "Stop: Success"
-    else:
-        #Antwort als Stream ausgeben... wenn Textantwort gefordert
-        history[-1][1] = ""
-        for character in result:
-            history[-1][1] += character
-            time.sleep(0.03)
-            yield history, "Generating"
-            if shared_state.interrupted:
-                shared_state.recover()
-                try:
-                    yield history, "Stop: Success"
-                    return
-                except:
-                    pass
+    #Antwort als Stream ausgeben... wenn Textantwort gefordert
+    history[-1][1] = ""
+    for character in result:
+        history[-1][1] += character
+        time.sleep(0.03)
+        yield history, "Generating"
+        if shared_state.interrupted:
+            shared_state.recover()
+            try:
+                yield history, "Stop: Success"
+                return
+            except:
+                pass
         
 ################################################
 #GUI