Update app.py

app.py

@@ -390,17 +390,11 @@ def invoke (prompt, file, history, rag_option, model_option, openai_api_key, k=3
         ###########################
         if (model_option == "OpenAI"):
             #Anfrage an OpenAI ----------------------------
-            if (prompt.find('Bild zeichnen') != -1):
+            if (prompt.find('zeichnen') != -1):
                 #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] = file #image
-                print("history......................")
-                print(history)
-                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)
@@ -429,26 +423,37 @@ 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: ...........")
-            result = llm_chain(llm, history_text_und_prompt)
+            if (prompt.find('zeichnen') != -1):
+                result = response.content
+            else:
+                result = llm_chain(llm, history_text_und_prompt)
             
 
     except Exception as e:
         raise gr.Error(e)
 
 
-    #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
+    if (prompt.find('zeichnen') != -1):
+        #Bild ausgeben
+        image = Image.open(io.BytesIO(result))
+        history[-1][1] = file #image
+        print("history......................")
+        print(history)
+        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
         
 ################################################
 #GUI