Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import re
 from http import HTTPStatus
-from typing import Dict, List, Optional, Tuple, TypedDict
+from typing import Dict, List, Optional, Tuple
 import base64
 import dashscope
 import gradio as gr
@@ -11,17 +11,37 @@ import modelscope_studio.components.base as ms
 import modelscope_studio.components.legacy as legacy
 import modelscope_studio.components.antd as antd
 
-# Type definitions
-History = List[Tuple[str, str]]
-Messages = List[Dict[str, str]]
+# Define system prompt
+SystemPrompt = """You are an AI assistant specialized in creating web applications and tools. 
+Generate complete, runnable code based on user requirements with modern best practices."""
 
-class SystemConfig(TypedDict):
-    system: str
+# Define demo list
+DEMO_LIST = [
+    {
+        "title": "Web Application",
+        "description": "Create a responsive web application with modern UI",
+        "icon": "🌐"
+    },
+    {
+        "title": "Data Visualization",
+        "description": "Build an interactive data visualization dashboard",
+        "icon": "📊"
+    },
+    {
+        "title": "Automation Tool",
+        "description": "Generate a Python automation tool",
+        "icon": "🛠️"
+    }
+]
 
 # Initialize DashScope
 YOUR_API_TOKEN = os.getenv('YOUR_API_TOKEN')
 dashscope.api_key = YOUR_API_TOKEN
 
+# Type definitions
+History = List[Tuple[str, str]]
+Messages = List[Dict[str, str]]
+
 def history_to_messages(history: History, system: str) -> Messages:
     messages = [{'role': Role.SYSTEM, 'content': system}]
     for h in history:
@@ -36,25 +56,25 @@ def messages_to_history(messages: Messages) -> History:
         history.append([q['content'], r['content']])
     return history
 
-def remove_code_block(text: str) -> str:
+def remove_code_block(text):
     pattern = r'```html\n(.+?)\n```'
     match = re.search(pattern, text, re.DOTALL)
     if match:
         return match.group(1).strip()
     return text.strip()
 
-def history_render(history: History) -> tuple[gr.update, History]:
+def history_render(history: History):
     return gr.update(open=True), history
 
-def clear_history() -> List:
+def clear_history():
     return []
 
-def send_to_sandbox(code: str) -> str:
+def send_to_sandbox(code):
     encoded_html = base64.b64encode(code.encode('utf-8')).decode('utf-8')
     data_uri = f"data:text/html;charset=utf-8;base64,{encoded_html}"
     return f"<iframe src=\"{data_uri}\" width=\"100%\" height=\"920px\"></iframe>"
 
-def demo_card_click(e: gr.EventData) -> str:
+def demo_card_click(e: gr.EventData):
     index = e._data['component']['index']
     return DEMO_LIST[index]['description']
 
@@ -64,7 +84,13 @@ with gr.Blocks(css="app.css") as demo:
     setting = gr.State({"system": SystemPrompt})
     
     with ms.Application() as app:
-        with antd.ConfigProvider():
+        with antd.ConfigProvider(theme={
+            "token": {
+                "colorPrimary": "#6366f1",
+                "borderRadius": 8,
+                "fontSize": 16
+            }
+        }):
             with antd.Row(gutter=[32, 12]) as layout:
                 # Left Column
                 with antd.Col(span=24, md=8):
@@ -94,8 +120,10 @@ with gr.Blocks(css="app.css") as demo:
                                     antd.CardMeta()
                                 demoCard.click(demo_card_click, outputs=[input])
                         
+                        antd.Divider("Settings")
+                        
                         with antd.Flex(gap="small", wrap=True):
-                            settingPromptBtn = antd.Button("⚙️ Settings", type="default")
+                            settingPromptBtn = antd.Button("⚙️ System Prompt", type="default")
                             codeBtn = antd.Button("🧑‍💻 View Code", type="default")
                             historyBtn = antd.Button("📜 History", type="default")
                 
@@ -194,8 +222,6 @@ with gr.Blocks(css="app.css") as demo:
         outputs=[code_output, history, sandbox, state_tab, code_drawer]
     )
     
-    clear_btn.click(clear_history, outputs=[history])
-    
     settingPromptBtn.click(
         lambda: gr.update(open=True),
         outputs=[system_prompt_modal]
@@ -206,11 +232,14 @@ with gr.Blocks(css="app.css") as demo:
         inputs=[systemPromptInput],
         outputs=[setting, system_prompt_modal]
     )
+    
+    clear_btn.click(clear_history, outputs=[history])
 
 # Launch the app
 if __name__ == "__main__":
     demo.queue(concurrency_count=20).launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=True
+        share=True,
+        ssr_mode=False
     )