Update app.py

app.py

@@ -1,9 +1,20 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
+from datetime import datetime
+
 
 client = InferenceClient("EvanZhouDev/open-genmoji", token=os.getenv("HUGGINGFACE_API_TOKEN"))
 llm = InferenceClient("Qwen/Qwen2.5-72B-Instruct")
+try:
+    os.mkdir("outputs")
+except FileExistsError:
+    print("Folder already exists.")
+
+try:
+    os.mkdir("outputs/images")
+except FileExistsError:
+    print("Folder already exists.")
 
 # Define the process function that takes a text prompt and returns an image
 def process(prompt):
@@ -14,10 +25,14 @@ def process(prompt):
     ]
 
     completion = llm.chat_completion(messages, max_tokens=100)
-    response = completion.get("choices")[0].get("message").get("content").replace("```", "").replace("\n", "")
 
+    response = completion.get("choices")[0].get("message").get("content").replace("```", "").replace("\n", "")
     print(response)
+    time = str(datetime.now())
     image = client.text_to_image(response)
+    image.save("outputs/images/" + time)
+    with open("outputs/" + time + ".json", "wb") as f:
+        f.write({"prompt": prompt, "refined_prompt": response, "image": "outputs/images/" + time})
     return image
 
 # Create a Gradio Blocks app