Spaces:
Running
Running
<script lang="ts"> | |
import hljs from 'highlight.js'; | |
import { loadPyodide } from 'pyodide'; | |
import mermaid from 'mermaid'; | |
import { v4 as uuidv4 } from 'uuid'; | |
import { getContext, getAllContexts, onMount, tick, createEventDispatcher } from 'svelte'; | |
import { copyToClipboard } from '$lib/utils'; | |
import 'highlight.js/styles/github-dark.min.css'; | |
import PyodideWorker from '$lib/workers/pyodide.worker?worker'; | |
import CodeEditor from '$lib/components/common/CodeEditor.svelte'; | |
import SvgPanZoom from '$lib/components/common/SVGPanZoom.svelte'; | |
const i18n = getContext('i18n'); | |
const dispatch = createEventDispatcher(); | |
export let id = ''; | |
export let save = false; | |
export let run = true; | |
export let token; | |
export let lang = ''; | |
export let code = ''; | |
export let className = 'my-2'; | |
export let editorClassName = ''; | |
export let stickyButtonsClassName = 'top-8'; | |
let _code = ''; | |
$: if (code) { | |
updateCode(); | |
} | |
const updateCode = () => { | |
_code = code; | |
}; | |
let _token = null; | |
let mermaidHtml = null; | |
let highlightedCode = null; | |
let executing = false; | |
let stdout = null; | |
let stderr = null; | |
let result = null; | |
let copied = false; | |
let saved = false; | |
const saveCode = () => { | |
saved = true; | |
code = _code; | |
dispatch('save', code); | |
setTimeout(() => { | |
saved = false; | |
}, 1000); | |
}; | |
const copyCode = async () => { | |
copied = true; | |
await copyToClipboard(code); | |
setTimeout(() => { | |
copied = false; | |
}, 1000); | |
}; | |
const checkPythonCode = (str) => { | |
// Check if the string contains typical Python syntax characters | |
const pythonSyntax = [ | |
'def ', | |
'else:', | |
'elif ', | |
'try:', | |
'except:', | |
'finally:', | |
'yield ', | |
'lambda ', | |
'assert ', | |
'nonlocal ', | |
'del ', | |
'True', | |
'False', | |
'None', | |
' and ', | |
' or ', | |
' not ', | |
' in ', | |
' is ', | |
' with ' | |
]; | |
for (let syntax of pythonSyntax) { | |
if (str.includes(syntax)) { | |
return true; | |
} | |
} | |
// If none of the above conditions met, it's probably not Python code | |
return false; | |
}; | |
const executePython = async (code) => { | |
if (!code.includes('input') && !code.includes('matplotlib')) { | |
executePythonAsWorker(code); | |
} else { | |
result = null; | |
stdout = null; | |
stderr = null; | |
executing = true; | |
document.pyodideMplTarget = document.getElementById(`plt-canvas-${id}`); | |
let pyodide = await loadPyodide({ | |
indexURL: '/pyodide/', | |
stdout: (text) => { | |
console.log('Python output:', text); | |
if (stdout) { | |
stdout += `${text}\n`; | |
} else { | |
stdout = `${text}\n`; | |
} | |
}, | |
stderr: (text) => { | |
console.log('An error occurred:', text); | |
if (stderr) { | |
stderr += `${text}\n`; | |
} else { | |
stderr = `${text}\n`; | |
} | |
}, | |
packages: ['micropip'] | |
}); | |
try { | |
const micropip = pyodide.pyimport('micropip'); | |
// await micropip.set_index_urls('https://pypi.org/pypi/{package_name}/json'); | |
let packages = [ | |
code.includes('requests') ? 'requests' : null, | |
code.includes('bs4') ? 'beautifulsoup4' : null, | |
code.includes('numpy') ? 'numpy' : null, | |
code.includes('pandas') ? 'pandas' : null, | |
code.includes('matplotlib') ? 'matplotlib' : null, | |
code.includes('sklearn') ? 'scikit-learn' : null, | |
code.includes('scipy') ? 'scipy' : null, | |
code.includes('re') ? 'regex' : null, | |
code.includes('seaborn') ? 'seaborn' : null | |
].filter(Boolean); | |
console.log(packages); | |
await micropip.install(packages); | |
result = await pyodide.runPythonAsync(`from js import prompt | |
def input(p): | |
return prompt(p) | |
__builtins__.input = input`); | |
result = await pyodide.runPython(code); | |
if (!result) { | |
result = '[NO OUTPUT]'; | |
} | |
console.log(result); | |
console.log(stdout); | |
console.log(stderr); | |
const pltCanvasElement = document.getElementById(`plt-canvas-${id}`); | |
if (pltCanvasElement?.innerHTML !== '') { | |
pltCanvasElement.classList.add('pt-4'); | |
} | |
} catch (error) { | |
console.error('Error:', error); | |
stderr = error; | |
} | |
executing = false; | |
} | |
}; | |
const executePythonAsWorker = async (code) => { | |
result = null; | |
stdout = null; | |
stderr = null; | |
executing = true; | |
let packages = [ | |
code.includes('requests') ? 'requests' : null, | |
code.includes('bs4') ? 'beautifulsoup4' : null, | |
code.includes('numpy') ? 'numpy' : null, | |
code.includes('pandas') ? 'pandas' : null, | |
code.includes('sklearn') ? 'scikit-learn' : null, | |
code.includes('scipy') ? 'scipy' : null, | |
code.includes('re') ? 'regex' : null, | |
code.includes('seaborn') ? 'seaborn' : null | |
].filter(Boolean); | |
console.log(packages); | |
const pyodideWorker = new PyodideWorker(); | |
pyodideWorker.postMessage({ | |
id: id, | |
code: code, | |
packages: packages | |
}); | |
setTimeout(() => { | |
if (executing) { | |
executing = false; | |
stderr = 'Execution Time Limit Exceeded'; | |
pyodideWorker.terminate(); | |
} | |
}, 60000); | |
pyodideWorker.onmessage = (event) => { | |
console.log('pyodideWorker.onmessage', event); | |
const { id, ...data } = event.data; | |
console.log(id, data); | |
data['stdout'] && (stdout = data['stdout']); | |
data['stderr'] && (stderr = data['stderr']); | |
data['result'] && (result = data['result']); | |
executing = false; | |
}; | |
pyodideWorker.onerror = (event) => { | |
console.log('pyodideWorker.onerror', event); | |
executing = false; | |
}; | |
}; | |
let debounceTimeout; | |
const drawMermaidDiagram = async () => { | |
try { | |
if (await mermaid.parse(code)) { | |
const { svg } = await mermaid.render(`mermaid-${uuidv4()}`, code); | |
mermaidHtml = svg; | |
} | |
} catch (error) { | |
console.log('Error:', error); | |
} | |
}; | |
const render = async () => { | |
if (lang === 'mermaid' && (token?.raw ?? '').slice(-4).includes('```')) { | |
(async () => { | |
await drawMermaidDiagram(); | |
})(); | |
} | |
}; | |
$: if (token) { | |
if (JSON.stringify(token) !== JSON.stringify(_token)) { | |
_token = token; | |
} | |
} | |
$: if (_token) { | |
render(); | |
} | |
$: dispatch('code', { lang, code }); | |
onMount(async () => { | |
console.log('codeblock', lang, code); | |
if (lang) { | |
dispatch('code', { lang, code }); | |
} | |
if (document.documentElement.classList.contains('dark')) { | |
mermaid.initialize({ | |
startOnLoad: true, | |
theme: 'dark', | |
securityLevel: 'loose' | |
}); | |
} else { | |
mermaid.initialize({ | |
startOnLoad: true, | |
theme: 'default', | |
securityLevel: 'loose' | |
}); | |
} | |
}); | |
</script> | |
<div> | |
<div class="relative {className} flex flex-col rounded-lg" dir="ltr"> | |
{#if lang === 'mermaid'} | |
{#if mermaidHtml} | |
<SvgPanZoom | |
className=" border border-gray-50 dark:border-gray-850 rounded-lg max-h-fit overflow-hidden" | |
svg={mermaidHtml} | |
content={_token.text} | |
/> | |
{:else} | |
<pre class="mermaid">{code}</pre> | |
{/if} | |
{:else} | |
<div class="text-text-300 absolute pl-4 py-1.5 text-xs font-medium dark:text-white"> | |
{lang} | |
</div> | |
<div | |
class="sticky {stickyButtonsClassName} mb-1 py-1 pr-2.5 flex items-center justify-end z-10 text-xs text-black dark:text-white" | |
> | |
<div class="flex items-center gap-0.5 translate-y-[1px]"> | |
{#if lang.toLowerCase() === 'python' || lang.toLowerCase() === 'py' || (lang === '' && checkPythonCode(code))} | |
{#if executing} | |
<div class="run-code-button bg-none border-none p-1 cursor-not-allowed">Running</div> | |
{:else if run} | |
<button | |
class="run-code-button bg-none border-none bg-gray-50 hover:bg-gray-100 dark:bg-gray-850 dark:hover:bg-gray-800 transition rounded-md px-1.5 py-0.5" | |
on:click={async () => { | |
code = _code; | |
await tick(); | |
executePython(code); | |
}}>{$i18n.t('Run')}</button | |
> | |
{/if} | |
{/if} | |
{#if save} | |
<button | |
class="save-code-button bg-none border-none bg-gray-50 hover:bg-gray-100 dark:bg-gray-850 dark:hover:bg-gray-800 transition rounded-md px-1.5 py-0.5" | |
on:click={saveCode} | |
> | |
{saved ? $i18n.t('Saved') : $i18n.t('Save')} | |
</button> | |
{/if} | |
<button | |
class="copy-code-button bg-none border-none bg-gray-50 hover:bg-gray-100 dark:bg-gray-850 dark:hover:bg-gray-800 transition rounded-md px-1.5 py-0.5" | |
on:click={copyCode}>{copied ? $i18n.t('Copied') : $i18n.t('Copy')}</button | |
> | |
</div> | |
</div> | |
<div | |
class="language-{lang} rounded-t-lg -mt-8 {editorClassName | |
? editorClassName | |
: executing || stdout || stderr || result | |
? '' | |
: 'rounded-b-lg'} overflow-hidden" | |
> | |
<div class=" pt-7 bg-gray-50 dark:bg-gray-850"></div> | |
<CodeEditor | |
value={code} | |
{id} | |
{lang} | |
on:save={() => { | |
saveCode(); | |
}} | |
on:change={(e) => { | |
_code = e.detail.value; | |
}} | |
/> | |
</div> | |
<div | |
id="plt-canvas-{id}" | |
class="bg-[#202123] text-white max-w-full overflow-x-auto scrollbar-hidden" | |
/> | |
{#if executing} | |
<div class="bg-[#202123] text-white px-4 py-4 rounded-b-lg"> | |
<div class=" text-gray-500 text-xs mb-1">STDOUT/STDERR</div> | |
<div class="text-sm">Running...</div> | |
</div> | |
{:else if stdout || stderr || result} | |
<div class="bg-[#202123] text-white px-4 py-4 rounded-b-lg"> | |
<div class=" text-gray-500 text-xs mb-1">STDOUT/STDERR</div> | |
<div class="text-sm">{stdout || stderr || result}</div> | |
</div> | |
{/if} | |
{/if} | |
</div> | |
</div> | |