Update app.py

app.py

@@ -143,13 +143,11 @@ def respond(message, history):
             elif json_data["name"] == "image_generation":
                 query = json_data["arguments"]["query"]
                 gr.Info("Generating Image, Please wait 10 sec...")
-                seed = random.randint(1, 99999)
-                image = f"![](https://image.pollinations.ai/prompt/{message_text}{query}?seed={seed}&nologo=True)"
-                image = image.replace("\\n", "")
-                image = image.replace(" ", "%20")
-                yield image
-                time.sleep(8)
-                gr.Info("We are going to Update Our Image Generation Engine to more powerful ones in Next Update. ThankYou")
+                client = InferenceClient("stabilityai/stable-diffusion-3-medium-diffusers")
+                seed = random.randint(0.999999)
+                image = client.text_to_image(message_text, negative_prompt=f"{seed}")
+                yield gr.Image(image)
+                gr.Info("We are going to mor upgrade image generator in next update")
             elif json_data["name"] == "image_qna":
                 inputs = llava(message, history)
                 streamer = TextIteratorStreamer(processor, skip_prompt=True, **{"skip_special_tokens": True})