Spaces:
Runtime error
Runtime error
File size: 9,458 Bytes
3878434 7f20e27 3878434 7f20e27 3878434 7f20e27 3878434 147e798 3878434 7f20e27 3878434 7f20e27 a26a4c4 3878434 7f20e27 3878434 a26a4c4 7f20e27 a26a4c4 7f20e27 115f611 147e798 a26a4c4 115f611 3878434 7f20e27 3878434 115f611 3878434 7f20e27 115f611 3878434 115f611 7f20e27 3878434 7f20e27 3878434 115f611 3878434 7f20e27 3878434 115f611 3878434 7f20e27 3878434 a26a4c4 7f20e27 a26a4c4 7f20e27 696fb43 3878434 7f20e27 3878434 7f20e27 147e798 a26a4c4 3878434 7f20e27 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 |
import os
import re
from http import HTTPStatus
from typing import Dict, List, Optional, Tuple
import base64
import dashscope
import gradio as gr
from dashscope import Generation
from dashscope.api_entities.dashscope_response import Role
import modelscope_studio.components.base as ms
import modelscope_studio.components.legacy as legacy
import modelscope_studio.components.antd as antd
from config import DEMO_LIST, SystemPrompt
# --- Setup API key ---
YOUR_API_TOKEN = os.getenv('YOUR_API_TOKEN')
if not YOUR_API_TOKEN:
raise ValueError("Environment variable 'YOUR_API_TOKEN' not set.")
dashscope.api_key = YOUR_API_TOKEN
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:
messages.append({'role': Role.USER, 'content': h[0]})
messages.append({'role': Role.ASSISTANT, 'content': h[1]})
return messages
def messages_to_history(messages: Messages) -> Tuple[str, History]:
assert messages[0]['role'] == Role.SYSTEM
history = []
for q, r in zip(messages[1::2], messages[2::2]):
history.append([q['content'], r['content']])
return history
def remove_code_block(text):
try:
pattern = r'```html\n(.+?)\n```'
match = re.search(pattern, text, re.DOTALL)
return match.group(1).strip() if match else text.strip()
except Exception as e:
print("Error in remove_code_block:", e)
return text
def history_render(history: History):
return gr.update(open=True), history
def clear_history():
return []
def send_to_sandbox(code):
try:
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>"
except Exception as e:
print("Error in send_to_sandbox:", e)
return "<p>Error rendering code</p>"
def demo_card_click(e: gr.EventData):
try:
index = e._data['component']['index']
return DEMO_LIST[index]['description']
except Exception as e:
print("Error in demo_card_click:", e)
return ""
with gr.Blocks(css_paths="app.css") as demo:
history = gr.State([])
setting = gr.State({"system": SystemPrompt})
with ms.Application() as app:
with antd.ConfigProvider():
with antd.Row(gutter=[32, 12]):
with antd.Col(span=24, md=8):
with antd.Flex(vertical=True, gap="middle", wrap=True):
gr.HTML("""
<div class="left_header">
<img src="//img.alicdn.com/imgextra/i2/O1CN01KDhOma1DUo8oa7OIU_!!6000000000220-1-tps-240-240.gif" width="200px" />
<h1>Qwen2.5-Coder</h2>
</div>
""")
input = antd.InputTextarea(size="large", allow_clear=True, placeholder="Please enter what kind of application you want")
btn = antd.Button("send", type="primary", size="large")
clear_btn = antd.Button("clear history", type="default", size="large")
antd.Divider("examples")
with antd.Flex(gap="small", wrap=True):
for demo_item in DEMO_LIST:
with antd.Card(hoverable=True, as_item="card") as demoCard:
antd.CardMeta()
demoCard.click(demo_card_click, outputs=[input])
antd.Divider("setting")
with antd.Flex(gap="small", wrap=True):
settingPromptBtn = antd.Button("⚙️ set system Prompt", type="default")
codeBtn = antd.Button("🧑💻 view code", type="default")
historyBtn = antd.Button("📜 history", type="default")
with antd.Modal(open=False, title="set system Prompt", width="800px") as system_prompt_modal:
systemPromptInput = antd.InputTextarea(SystemPrompt, auto_size=True)
settingPromptBtn.click(lambda: gr.update(open=True), inputs=[], outputs=[system_prompt_modal])
system_prompt_modal.ok(lambda input: ({"system": input}, gr.update(open=False)), inputs=[systemPromptInput], outputs=[setting, system_prompt_modal])
system_prompt_modal.cancel(lambda: gr.update(open=False), outputs=[system_prompt_modal])
with antd.Drawer(open=False, title="code", placement="left", width="750px") as code_drawer:
code_output = legacy.Markdown()
codeBtn.click(lambda: gr.update(open=True), inputs=[], outputs=[code_drawer])
code_drawer.close(lambda: gr.update(open=False), inputs=[], outputs=[code_drawer])
with antd.Drawer(open=False, title="history", placement="left", width="900px") as history_drawer:
history_output = legacy.Chatbot(show_label=False, flushing=False, height=960, elem_classes="history_chatbot")
historyBtn.click(history_render, inputs=[history], outputs=[history_drawer, history_output])
history_drawer.close(lambda: gr.update(open=False), inputs=[], outputs=[history_drawer])
with antd.Col(span=24, md=16):
with ms.Div(elem_classes="right_panel"):
gr.HTML('<div class="render_header"><span class="header_btn"></span><span class="header_btn"></span><span class="header_btn"></span></div>')
with antd.Tabs(active_key="empty", render_tab_bar="() => null") as state_tab:
with antd.Tabs.Item(key="empty"):
antd.Empty(description="empty input", elem_classes="right_content")
with antd.Tabs.Item(key="loading"):
antd.Spin(True, tip="coding...", size="large", elem_classes="right_content")
with antd.Tabs.Item(key="render"):
sandbox = gr.HTML(elem_classes="html_content")
def generation_code(query: Optional[str], _setting: Dict[str, str], _history: Optional[History]):
try:
if query is None:
query = ''
if _history is None:
_history = []
messages = history_to_messages(_history, _setting['system'])
messages.append({'role': Role.USER, 'content': query})
print("Sending messages to dashscope:", messages)
gen = Generation.call(
model="qwen2.5-coder-32b-instruct",
messages=messages,
result_format='message',
stream=True
)
for response in gen:
print("Response received:", response)
if response.status_code == HTTPStatus.OK:
role = response.output.choices[0].message.role
content = response.output.choices[0].message.content
print("Content:", content)
if response.output.choices[0].finish_reason == 'stop':
_history = messages_to_history(messages + [{'role': role, 'content': content}])
yield {
code_output: content,
history: _history,
sandbox: send_to_sandbox(remove_code_block(content)),
state_tab: gr.update(active_key="render"),
code_drawer: gr.update(open=False),
}
else:
yield {
code_output: content,
state_tab: gr.update(active_key="loading"),
code_drawer: gr.update(open=True),
}
else:
raise ValueError(
f"Request ID: {response.request_id}, Status code: {response.status_code}, "
f"Error Code: {response.code}, Error Message: {response.message}"
)
except Exception as e:
print("Error in generation_code:", e)
yield {
code_output: f"❌ Error: {e}",
sandbox: "",
state_tab: gr.update(active_key="empty")
}
btn.click(generation_code,
inputs=[input, setting, history],
outputs=[code_output, history, sandbox, state_tab, code_drawer])
clear_btn.click(clear_history, inputs=[], outputs=[history])
if __name__ == "__main__":
demo.queue(default_concurrency_limit=20).launch(ssr_mode=False)
|