Update app.py

app.py

@@ -225,35 +225,26 @@ def generate_auswahl(prompt, file, chatbot, history, rag_option, model_option, o
         
 ##################################################
 #zu einem Text-Prompt ein Bild via Stable Diffusion generieren
-def generate_bild(prompt, chatbot,  temperature=0.5, max_new_tokens=4048,top_p=0.6, repetition_penalty=1.3):
-    """
-    print("Bild Erzeugung HF..............................")
-    #Bild nach Anweisung zeichnen und in History darstellen...
-    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)) 
-    image_64 = umwandeln_fuer_anzeige(image)
-    chatbot[-1][1]= "<img src='data:image/png;base64,{0}'/>".format(base64.b64encode(image_64).decode('utf-8'))
-    return chatbot, "Success"
-    """
-    print("Bild Erzeugung DallE..............................")
-    client = OpenAI()
-    #als Format ginge auch 'url', n - Anz. der erzeugten Bilder
-    response = client.images.generate(model="dall-e-3",prompt=prompt,size="1024x1024",quality="standard",n=1, response_format='b64_json') 
-    print("response.........................")
-    print(response)
-    print("data........................")
-    print(response.data[0].b64_json)
-    #with open(image_path, "rb") as image_file:
-        #return base64.b64encode(image_file.read()).decode('utf-8')
+def generate_bild(prompt, chatbot, model_option_zeichnen, temperature=0.5, max_new_tokens=4048,top_p=0.6, repetition_penalty=1.3):
+    if (model_option_zeichnen == "HugginfFace"):
+        print("Bild Erzeugung HF..............................")
+        #Bild nach Anweisung zeichnen und in History darstellen...
+        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)) 
+        image_64 = umwandeln_fuer_anzeige(image)
+        chatbot[-1][1]= "<img src='data:image/png;base64,{0}'/>".format(base64.b64encode(image_64).decode('utf-8'))
+    else:     
+        print("Bild Erzeugung DallE..............................")
+        client = OpenAI()
+        #als Format ginge auch 'url', n - Anz. der erzeugten Bilder
+        response = client.images.generate(model="dall-e-3",prompt=prompt,size="1024x1024",quality="standard",n=1, response_format='b64_json') 
+        #chatbot[-1][1]= "<img src='data:image/png;base64,{0}'/>".format(base64.b64encode(image_64).decode('utf-8'))
+        chatbot[-1][1] = "<img src='data:image/png;base64,{0}'/>".format(response.data[0].b64_json)
     
-    #image = Image.open(io.BytesIO(response)) 
-    #image_64 = umwandeln_fuer_anzeige(image)
-    #chatbot[-1][1]= "<img src='data:image/png;base64,{0}'/>".format(base64.b64encode(image_64).decode('utf-8'))
-    chatbot[-1][1] = "<img src='data:image/png;base64,{0}'/>".format(response.data[0].b64_json)
     return chatbot, "Success"
     
 
@@ -484,7 +475,7 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
         with gr.Row():
             with gr.Column(scale=5):
                 with gr.Row():
-                    chatbot_bild = gr.Chatbot(elem_id="li-zeichnen")
+                    chatbot_bild = gr.Chatbot(elem_id="li-zeichnen", ,show_copy_button=True, show_share_button=True)
                 with gr.Row():
                     with gr.Column(scale=12):
                         user_input2 = gr.Textbox(
@@ -498,6 +489,12 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
                 with gr.Row():
                     emptyBtn2 = gr.ClearButton([user_input, chatbot_bild], value="🧹 Neue Session", scale=10)  
                     #additional_inputs_accordion = gr.Accordion(label="Weitere Eingaben...",  open=False)
+             with gr.Column():
+                with gr.Column(min_width=50, scale=1):
+                    with gr.Tab(label="Parameter Einstellung"):
+                        #gr.Markdown("# Parameters")
+                        model_option_zeichnen = gr.Radio(["OpenAI(DallE)", "HuggingFace"], label="Modellauswahl", value = "HugginfFace")
+
 
     gr.Markdown(description)        
     
@@ -553,9 +550,10 @@ with gr.Blocks(css=customCSS, theme=small_and_beautiful_theme) as demo:
         inputs=[
             user_question2,
             chatbot_bild,
+            model_option_zeichnen,
             #additional_inputs,
         ],
-        outputs=[chatbot_bild, status_display2], #[chatbot, history,  status_display]
+        outputs=[chatbot_bild,  status_display2], #[chatbot, history,  status_display]
         show_progress=True,
     )
     transfer_input_args2 = dict(