generate add image

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from datetime import datetime, timedelta
 
 from gemini_api import model_api, sentiment, category, ord_num, NO_ORDER, NO_ITEM, food_return, cloth_return, item_identy, item_match, generate_add
-from openai_api import model_api as openai_model_api, sentiment as openai_sentiment
+from openai_api import model_api as openai_model_api, sentiment as openai_sentiment, image_gen
 
 cust_qry_resp = {"senti":"", "cat":"", "num":""}
 
@@ -124,8 +124,11 @@ with gr.Blocks(title="Customer Support Assistant",
     age_grp = ip_arr[2]
     if ord_item is not None:
       ret_add = generate_add(ord_item, theme, age_grp)
+      image_url = image_gen(f"{ord_item} {theme} {age_grp}")
       with gr.Row():
         gr.Textbox(type="text", label="Genarated Add", value=ret_add, interactive=True)
+      with gr.Row():
+        gr.Image(height=1024, width=1024, value=image_url)
   
 if __name__ == "__main__":
   app.launch(server_name="0.0.0.0")

openai_api.py

@@ -37,4 +37,17 @@ def sentiment(text):
     # Extract the sentiment from the response
     sentiment = response.choices[0].message.content.strip()
 
-    return sentiment
+    return sentiment
+
+def image_gen(text):
+    print(text)    
+    response = client.images.generate(
+      model="dall-e-3",
+      prompt= text,
+      size="1024x1024",
+      quality="standard",
+      n=1,
+    )
+
+    image_url = response.data[0].url
+    return image_url