feat: add writing helper tab

app.py

@@ -12,7 +12,7 @@ import modelscope_studio.components.pro as pro
 from modelscope_studio.components.pro.chatbot import (
     ChatbotActionConfig, ChatbotBotConfig, ChatbotMarkdownConfig,
     ChatbotPromptsConfig, ChatbotUserConfig, ChatbotWelcomeConfig)
-from config import DEFAULT_PROMPTS, EXAMPLES, SystemPrompt
+from config import DEFAULT_PROMPTS, EXAMPLES, CODE_SYSTEM_PROMPT, WRITING_DEFAULT_PROMPTS
 import re
 
 MODEL_VERSION = os.environ['MODEL_VERSION']
@@ -41,8 +41,7 @@ def retry(chatbot_value, e: gr.EventData):
     index = e._data["payload"][0]["index"]
     chatbot_value = chatbot_value[:index]
 
-    yield gr.update(loading=True), gr.update(value=chatbot_value), gr.update(
-        disabled=True)
+    yield gr.update(value=chatbot_value)
     for chunk in submit(None, chatbot_value):
         yield chunk
 
@@ -161,11 +160,7 @@ def submit(sender_value, chatbot_value):
         "status": "pending"
     })
 
-    yield {
-        sender: gr.update(value=None, loading=True),
-        clear_btn: gr.update(disabled=True),
-        chatbot: gr.update(value=chatbot_value)
-    }
+    yield gr.update(value=chatbot_value)
 
     try:
         data = {
@@ -249,7 +244,7 @@ def submit(sender_value, chatbot_value):
                                 "status": "done"
                             }
 
-                    yield {chatbot: gr.update(value=chatbot_value)}
+                    yield gr.update(value=chatbot_value)
 
                 elif 'message' in choice:
                     message_data = choice['message']
@@ -279,26 +274,18 @@ def submit(sender_value, chatbot_value):
                         }
 
                     chatbot_value[-1]["loading"] = False
-                    yield {chatbot: gr.update(value=chatbot_value)}
+                    yield gr.update(value=chatbot_value)
 
         chatbot_value[-1]["footer"] = "{:.2f}s".format(time.time() -
                                                        start_time)
         chatbot_value[-1]["status"] = "done"
-        yield {
-            clear_btn: gr.update(disabled=False),
-            sender: gr.update(loading=False),
-            chatbot: gr.update(value=chatbot_value),
-        }
+        yield gr.update(value=chatbot_value)
 
     except Exception as e:
         chatbot_value[-1]["loading"] = False
         chatbot_value[-1]["status"] = "done"
         chatbot_value[-1]["content"] = "Request failed, please try again."
-        yield {
-            clear_btn: gr.update(disabled=False),
-            sender: gr.update(loading=False),
-            chatbot: gr.update(value=chatbot_value),
-        }
+        yield gr.update(value=chatbot_value)
         raise e
 
 
@@ -398,7 +385,7 @@ def generate_code(query: str):
     print("Starting code generation with query:", query)
     messages = [{
         'role': 'system',
-        'content': SystemPrompt
+        'content': CODE_SYSTEM_PROMPT
     }, {
         'role': 'user',
         'content': query
@@ -769,62 +756,66 @@ def scroll_to_bottom():
     """
 
 
+def create_chatbot_tab(title, subtitle, prompts):
+    with antd.Flex(vertical=True,
+                   gap="middle",
+                   elem_style=dict(height="calc(100vh - 150px)")):
+        chatbot = pro.Chatbot(
+            elem_style=dict(flex=1, maxHeight=1200, height=0),
+            markdown_config=ChatbotMarkdownConfig(
+                allow_tags=["think"]),
+            welcome_config=ChatbotWelcomeConfig(
+                variant="borderless",
+                icon="./assets/minimax-logo.png",
+                title=title,
+                description="You can input text to get started.",
+                prompts=ChatbotPromptsConfig(
+                    title=subtitle,
+                    styles={
+                        "list": {
+                            "width": '100%',
+                        },
+                        "item": {
+                            "flex": 1,
+                        },
+                    },
+                    items=prompts)),
+            user_config=ChatbotUserConfig(actions=["copy", "edit"]),
+            bot_config=ChatbotBotConfig(
+                header=MODEL_NAME,
+                avatar="./assets/minimax-logo.png",
+                actions=["copy", "retry"]))
+
+        with antdx.Sender() as sender:
+            with ms.Slot("prefix"):
+                with antd.Button(value=None,
+                                 color="default",
+                                 variant="text") as clear_btn:
+                    with ms.Slot("icon"):
+                        antd.Icon("ClearOutlined")
+
+        clear_btn.click(fn=clear, outputs=[chatbot])
+        submit_event = sender.submit(
+            fn=submit,
+            inputs=[sender, chatbot],
+            outputs=[chatbot])
+        sender.cancel(fn=cancel,
+                      inputs=[chatbot],
+                      outputs=[chatbot, sender, clear_btn],
+                      cancels=[submit_event],
+                      queue=False)
+        chatbot.retry(fn=retry,
+                      inputs=[chatbot],
+                      outputs=[chatbot])
+        chatbot.welcome_prompt_select(fn=prompt_select,
+                                      outputs=[sender])
+
+
 with gr.Blocks(css=css) as demo, ms.Application(), antdx.XProvider(
 ), ms.AutoLoading():
     with antd.Tabs() as tabs:
         with antd.Tabs.Item(key="chat", label="Chatbot"):
-            with antd.Flex(vertical=True,
-                           gap="middle",
-                           elem_style=dict(height="calc(100vh - 150px)")):
-                chatbot = pro.Chatbot(
-                    elem_style=dict(flex=1, maxHeight=1200, height=0),
-                    markdown_config=ChatbotMarkdownConfig(
-                        allow_tags=["think"]),
-                    welcome_config=ChatbotWelcomeConfig(
-                        variant="borderless",
-                        icon="./assets/minimax-logo.png",
-                        title="Hello, I'm MiniMax-M1",
-                        description="You can input text to get started.",
-                        prompts=ChatbotPromptsConfig(
-                            title="How can I help you today?",
-                            styles={
-                                "list": {
-                                    "width": '100%',
-                                },
-                                "item": {
-                                    "flex": 1,
-                                },
-                            },
-                            items=DEFAULT_PROMPTS)),
-                    user_config=ChatbotUserConfig(actions=["copy", "edit"]),
-                    bot_config=ChatbotBotConfig(
-                        header=MODEL_NAME,
-                        avatar="./assets/minimax-logo.png",
-                        actions=["copy", "retry"]))
-
-                with antdx.Sender() as sender:
-                    with ms.Slot("prefix"):
-                        with antd.Button(value=None,
-                                         color="default",
-                                         variant="text") as clear_btn:
-                            with ms.Slot("icon"):
-                                antd.Icon("ClearOutlined")
-
-                clear_btn.click(fn=clear, outputs=[chatbot])
-                submit_event = sender.submit(
-                    fn=submit,
-                    inputs=[sender, chatbot],
-                    outputs=[sender, chatbot, clear_btn])
-                sender.cancel(fn=cancel,
-                              inputs=[chatbot],
-                              outputs=[chatbot, sender, clear_btn],
-                              cancels=[submit_event],
-                              queue=False)
-                chatbot.retry(fn=retry,
-                              inputs=[chatbot],
-                              outputs=[sender, chatbot, clear_btn])
-                chatbot.welcome_prompt_select(fn=prompt_select,
-                                              outputs=[sender])
+            create_chatbot_tab("Chatbot", "How can I help you today?", DEFAULT_PROMPTS)
 
         with antd.Tabs.Item(key="code", label="Code Playground (WebDev)"):
             with antd.Row(gutter=[32, 12],
@@ -927,6 +918,9 @@ with gr.Blocks(css=css) as demo, ms.Application(), antdx.XProvider(
                 fn=on_tab_change,
                 outputs=[output_tabs],
             )
+        
+        with antd.Tabs.Item(key="writing", label="Writing Helper"):
+            create_chatbot_tab("Writing Helper", "What do you want to write?", WRITING_DEFAULT_PROMPTS)
 
 if __name__ == '__main__':
     demo.queue(default_concurrency_limit=50).launch(ssr_mode=False)

config.py

@@ -38,7 +38,7 @@ DEFAULT_CODE = {
     """
 }
 
-SystemPrompt = """
+CODE_SYSTEM_PROMPT = """
 You are a web development engineer, writing web pages according to the instructions below. You are a powerful code editing assistant capable of writing code and creating artifacts in conversations with users, or modifying and updating existing artifacts as requested by users. 
 All code is written in a single code block to form a complete code file for display, without separating HTML and JavaScript code. An artifact refers to a runnable complete code snippet, you prefer to integrate and output such complete runnable code rather than breaking it down into several code blocks. For certain types of code, they can render graphical interfaces in a UI window. After generation, please check the code execution again to ensure there are no errors in the output.
 Output only the HTML, without any additional descriptive text. Make the UI looks modern and beautiful.
@@ -105,4 +105,33 @@ EXAMPLES = {
             "description": "Create an image filter tool where users can upload an image and apply filters like grayscale, blur, brightness, contrast, etc. Show real-time preview."
         }
     ]
-}
+}
+
+WRITING_SYSTEM_PROMPT = """
+You are a helpful assistant.
+"""
+
+WRITING_DEFAULT_PROMPTS = [
+    {
+        "label": "📚 Creative Narratives",
+        "children": [
+            {
+                "description": "Write a 3-minute fantasy short story titled 'Living Books of the Ancient Library.' Use a classic three-act structure, feature at least two vivid protagonists, and end with an open-ended twist"
+            },
+            {
+                "description": "Draft a 25-minute podcast script on 'Breakthroughs in Deep-Sea Exploration.' Include host intro banter, three interview segments (scientist, submersible pilot, environmentalist), smooth transitions, and a CTA recap."
+            }
+        ]
+    },
+    {
+        "label": "📊 Insight & Analysis",
+        "children": [
+            {
+                "description": "Write a 2000-word white paper for startup founders on 'Quantum-Computing-Driven Drug Discovery.' Cover market overview, core tech principles, risks/regs, and next-step recommendations"
+            },
+            {
+                "description": "Create a 1500-word Medium article titled 'Sustainable Travel: A Practical Guide.' Open with a personal hook, break down key tactics with examples, and finish with an actionable checklist."
+            }
+        ]
+    }
+]