Update app.py

app.py

@@ -361,8 +361,27 @@ 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,):
+
+def generate_auswahl(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,):
+     if (prompt.find('zeichnen') != -1):
+         generate_bild(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,):
+     else:
+         generate_text(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,):
+
+def generate_bild(prompt, history,):
+    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)) 
+        
+    history[-1][1] =  "<img src='data:image/png;base64,{0}'/>".format(b64encode(umwandeln_fuer_anzeige(image)).decode('utf-8'))   
+    print("history zeichnen......................")
+    print(history)
+    return history, "Fertig: Success"
+
+def generate_text (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
     print(splittet)
 
@@ -558,7 +577,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
 
     #Argumente für generate Funktion als Input
     predict_args = dict(
-        fn=invoke,
+        fn=generate_auswahl,
         inputs=[
             user_question,
             upload,