Update app.py

app.py

@@ -109,7 +109,7 @@ def respond(message, history):
         functions_metadata = [
             {"type": "function", "function": {"name": "web_search", "description": "Search query on google", "parameters": {"type": "object", "properties": {"query": {"type": "string", "description": "web search query"}}, "required": ["query"]}}},
             {"type": "function", "function": {"name": "general_query", "description": "Reply general query of USER", "parameters": {"type": "object", "properties": {"prompt": {"type": "string", "description": "A detailed prompt"}}, "required": ["prompt"]}}},
-            {"type": "function", "function": {"name": "image_generation", "description": "Generate image for user", "parameters": {"type": "object", "properties": {"query": {"type": "string", "description": "image generation prompt"}, "number_of_image": {"type": "integer", "description": "number of images to generate"}}, "required": ["query"]}}},
+            {"type": "function", "function": {"name": "image_generation", "description": "Generate image for user", "parameters": {"type": "object", "properties": {"query": {"type": "string", "description": "image generation prompt"}}, "required": ["query"]}}},
             {"type": "function", "function": {"name": "image_qna", "description": "Answer question asked by user related to image", "parameters": {"type": "object", "properties": {"query": {"type": "string", "description": "Question by user"}}, "required": ["query"]}}},
         ]
 
@@ -153,13 +153,16 @@ def respond(message, history):
                         yield output
             elif json_data["name"] == "image_generation":
                 query = json_data["arguments"]["query"]
+                gr.Info("Generating Image, Please wait 10 sec...")
+                yield "Generating Image, Please wait 10 sec..."
                 try:
-                    number_of_image = json_data["arguments"]["number_of_image"]
+                    image = image_gen(f"{str(query)}")
+                    yield gr.Image(image[1])
                 except:
-                    number_of_image = 1
-                gr.Info("Generating Image, Please wait 10 sec...")
-                image = image_gen(f"{str(query)}")
-                yield gr.Image(image[1])
+                    client_sd3 = InferenceClient("stabilityai/stable-diffusion-3-medium-diffusers")
+                    seed = random.randint(0,999999)
+                    image = client_sd3.text_to_image(query, negative_prompt=f"{seed}")
+                    yield gr.Image(image)
             elif json_data["name"] == "image_qna":
                 inputs = llava(message, history)
                 streamer = TextIteratorStreamer(processor, skip_prompt=True, **{"skip_special_tokens": True})