add stable-diffusion conversation example

examples/conversation_with_stablediffusion_model/README.md

@@ -0,0 +1,32 @@
+Launching Web UI with arguments: --api
+# Conversation with Stable-diffusion model
+
+This example will show how to program a multi-agent conversation in AgentScope.
+Complete code is in `conversation.py`, which set up a user agent and an
+assistant agent to have a conversation. When user input "exit", the
+conversation ends.
+You can modify the `sys_prompt` to change the role of assistant agent.
+```bash
+# Note: Set your api_key in conversation.py first
+python conversation.py
+```
+
+## Prerequisites
+
+You need to satisfy the following requirements to run this example:
+
+- Install Stable Diffusion Web UI by following the instructions at [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui).
+- Add the `--api` parameter to the startup options of the Stable Diffusion Web UI.
+- Ensure that your host can successfully access `http://127.0.0.1:7860/`(default) any other specified host and port you choose.
+- Install the latest version of AgentScope by
+ ```bash
+ git clone https://github.com/modelscope/agentscope.git
+ cd agentscope
+ pip install -e .
+ ```
+
+## Running the Example
+Run the example and input your questions.
+```bash
+python conversation_with_stablediffusion_model.py
+```

examples/conversation_with_stablediffusion_model/conversation_with_stablediffusion_model.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+"""A simple example for conversation between user and stable-diffusion agent."""
+import agentscope
+from agentscope.agents import DialogAgent
+from agentscope.agents.user_agent import UserAgent
+
+
+def main() -> None:
+ """A basic conversation demo"""
+
+ agentscope.init(
+ model_configs=[
+ {
+ "model_type": "sd_txt2img",
+ "config_name": "sd",
+ "options": {
+ "sd_model_checkpoint": "xxxxxx",
+ "CLIP_stop_at_last_layers": 2,
+ }, # global settings, for detailed parameters
+ # please refer to yourhost:yourpost/docs#/default/get_config_sdapi_v1_options_get
+ "generate_args": {
+ "steps": 50,
+ "n_iter": 1,
+ "override_settings": {
+ "CLIP_stop_at_last_layers": 3,
+ # settings effective only for this conversation
+ # The parameters are consistent with the global settings.
+ },
+ },
+ },
+ ],
+ project="txt2img-Agent Conversation",
+ save_api_invoke=True,
+ )
+
+ # Init two agents
+ dialog_agent = DialogAgent(
+ name="Assistant",
+ sys_prompt="high definition,dreamy", # replace by your desired image style prompts
+ model_config_name="sd", # replace by your model config name
+ )
+ user_agent = UserAgent()
+
+ # start the conversation between user and assistant
+ msg = None
+ while True:
+ msg = user_agent(msg)
+ if msg.content == "exit":
+ break
+ msg = dialog_agent(msg)
+
+
+if __name__ == "__main__":
+ main()