Update app.py

First image version

app.py

@@ -11,6 +11,10 @@ import queue
 import threading
 from glob import glob
 import copy
+import base64
+import json
+from PIL import Image
+from io import BytesIO
 
 load_dotenv(override=True)
 key = os.getenv('OPENAI_API_KEY')
@@ -130,7 +134,7 @@ def genUsageStats(do_reset=False):
 
 def new_conversation(user):
     clean_up(user)
-    return [None, [], None, []]
+    return [None, [], None, None]
 
 def updatePassword(txt):
     password = txt.lower().strip()
@@ -267,6 +271,10 @@ def final_clean_up(user):
         except:
             continue
 
+def delete_image(user):
+    fpath = dataDir + user + '.png'
+    if os.path.exists(fpath):
+        os.remove(fpath)
 
 def list_permanent_files():
     flist = os.listdir(dataDir)
@@ -284,6 +292,26 @@ def list_permanent_files():
     other_cnt = len(others)
     return (str(log_cnt), str(wav_cnt), str(other_cnt), str(others))
 
+def make_image(prompt, user, pwd):
+    user = user.lower().strip()
+    msg = 'Error: unable to create image.'
+    fpath = None
+    if user in unames and pwd == pwdList[unames.index(user)]:
+        response = client.images.generate(model='dall-e-2', prompt=prompt,size='512x512',
+               quality='standard', response_format='b64_json')
+        try:
+            image_data = response.data[0].b64_json 
+            image = Image.open(BytesIO(base64.b64decode(image_data)))
+            fpath = dataDir + user + '.png'
+            image.save(fpath)
+            msg = 'Image created!'
+        except:
+            return [gr.Image(value=None), msg]
+    else:
+        msg = 'Incorrect user name or password'
+        return [gr.Image(value=None), msg]
+    return [fpath, msg]
+
 
 with gr.Blocks() as demo:
     history = gr.State([])
@@ -406,13 +434,15 @@ with gr.Blocks() as demo:
         clear_button = gr.Button(value="Restart Conversation")
         # gpt_chooser=gr.Radio(choices=[("GPT-3.5","gpt-3.5-turbo"),("GPT-4o","gpt-4o-mini")],
         #                      value="gpt-3.5-turbo", label="GPT Model", interactive=True)
+        button_do_image = gr.Button(value='Make Image')
         submit_button = gr.Button(value="Submit Prompt/Question")
         speak_output = gr.Button(value="Speak Dialog", visible=False)
     prompt_window = gr.Textbox(label = "Prompt or Question")
     output_window = gr.Textbox(label = "Dialog")
+    image_window = gr.Image()
     submit_button.click(chat, inputs=[prompt_window, user_window, password, history, output_window, model],
                          outputs=[history, output_window, prompt_window, model])
-    clear_button.click(fn=new_conversation, inputs=user_window, outputs=[prompt_window, history, output_window])
+    clear_button.click(fn=new_conversation, inputs=user_window, outputs=[prompt_window, history, output_window, image_window])
     audio_widget.stop_recording(fn=transcribe, inputs=[user_window, password, audio_widget],
                                 outputs=[prompt_window])
     audio_widget.pause_recording(fn=pause_message, outputs=[prompt_window])
@@ -421,5 +451,7 @@ with gr.Blocks() as demo:
     audio_out.stop(fn=gen_output_audio, inputs=[q, user_window], outputs = [audio_out, q])
     speak_output.click(fn=initial_audio_output, inputs=[output_window, user_window], outputs=[audio_out, q])
     output_window.change(fn=set_speak_button, inputs=output_window,outputs=speak_output)
+    button_do_image.click(fn=make_image, inputs=[prompt_window,user_window, password],outputs=[image_window, output_window])
+    image_window.change(fn=delete_image, inputs=[user])
     # demo.unload(final_clean_up(user))
 demo.launch(share=True)