Final brush up

app.py

@@ -130,30 +130,44 @@ def chunk_code(
     except Exception as e:
         return str(e)
 
+examples_dict = {
+    "Python: Sweep's GiHub Actions log handler": ("https://raw.githubusercontent.com/sweepai/sweep/b267b613d4c706eaf959fe6789f11e9a856521d1/sweepai/handlers/on_check_suite.py", "python"),
+    "Typescript: LlamaIndex TS's BaseIndex abstract base class": ("https://raw.githubusercontent.com/run-llama/LlamaIndexTS/bfab1d407b7b390d76b3d7a1a1df0928e9f9ae11/packages/core/src/indices/BaseIndex.ts", "typescript"),
+    "Rust: Ruff's autofix code modification algorithm": ("https://raw.githubusercontent.com/astral-sh/ruff/main/crates/ruff/src/autofix/codemods.rs", "rust"),
+    "Go: Infisical's CLI's config manager": ("https://raw.githubusercontent.com/Infisical/infisical/de7bd27b4b48847c9ca7cd12d208225b06f170fe/cli/packages/util/config.go", "go")
+}
+
+default_key = "Python: GiHub Actions log handler"
+default_url, default_language = examples_dict[default_key]
+default_code = requests.get(default_url).text
+
 with gr.Blocks(css=css) as demo:
     gr.Markdown("## Code Chunking Demo")
-    gr.Markdown("Start typing below and the chunked output will automatically show up. Checkout how this algorithm works at https://docs.sweep.dev/blogs/chunking-2m-files and https://docs.sweep.dev/blogs/chunking-improvements. We also have interactive notebooks at https://github.com/sweepai/sweep/blob/main/notebooks/chunking.ipynb.")
-
-    default_file = "https://raw.githubusercontent.com/sweepai/sweep/b267b613d4c706eaf959fe6789f11e9a856521d1/sweepai/handlers/on_check_suite.py"
-    default_code = requests.get(default_file).text
+    gr.Markdown("Start typing below and the chunked output will automatically show up. Checkout how this algorithm works at https://docs.sweep.dev/blogs/chunking-2m-files and https://docs.sweep.dev/blogs/chunking-improvements or play with the notebook at https://github.com/sweepai/sweep/blob/main/notebooks/chunking.ipynb.")
 
     with gr.Row():
-        language = gr.Dropdown(["python", "javascript", "go", "ruby", "java", "php", "c", "cpp", "rust", "haskell"], label="Language", value="python")
+        language = gr.Dropdown(['python', 'javascript', 'typescript', 'rust', 'go', 'ruby', 'r', 'html', 'css', 'shell'], label="Language", value=default_language)
         max_chars = gr.Slider(10, 3000, 1500, label="Max Characters", step=10)
         coalesce = gr.Slider(0, 300, 100, label="Coalesce", step=10)
+    examples = gr.Dropdown(list(examples_dict.keys()), label="Examples", value=default_key, interactive=True)
     with gr.Row():
-        inp = gr.Code(placeholder="Enter the code here", label="Code to Chunk", language=language.value, lines=60, elem_classes="code_container", value=default_code)
-        out = gr.Code(label="Chunked Code", language=language.value, lines=60, value=chunk_code(default_code, language.value, max_chars.value, coalesce.value))
-
-    def update_language(inp, language, max_chars, coalesce):
-        return (
-            gr.update(language=language),
-            gr.update(language=language, value=chunk_code(inp.value, language, max_chars, coalesce))
-        )
-
-    language.change(fn=update_language, inputs=[inp, language, max_chars, coalesce], outputs=[inp, out])
-    max_chars.change(fn=chunk_code, inputs=[inp, language, max_chars, coalesce], outputs=out)
-    coalesce.change(fn=chunk_code, inputs=[inp, language, max_chars, coalesce], outputs=out)
-    inp.change(fn=chunk_code, inputs=[inp, language, max_chars, coalesce], outputs=out)
+        input_code = gr.Code(label="Input Code", language=language.value, lines=60, elem_classes="code_container", value=default_code)
+        output_code = gr.Code(label="Chunked Code", language=language.value, lines=60, value=chunk_code(default_code, language.value, max_chars.value, coalesce.value))
+    
+    def update_examples(examples):
+        url, language = examples_dict[examples]
+        code = requests.get(url).text
+        return gr.Code.update(language=language, value=code), gr.Code.update(language=language, value=chunk_code(code, language, max_chars.value, coalesce.value)), language
+    
+    examples.change(fn=update_examples, inputs=examples, outputs=[input_code, output_code, language])
+
+    def update_language(language):
+        return gr.Code.update(language=language), gr.Code.update(language=language)
+
+    language.change(fn=update_language, inputs=language, outputs=[input_code, output_code]) \
+        .then(fn=chunk_code, inputs=[input_code, language, max_chars, coalesce], outputs=output_code)
+    max_chars.change(fn=chunk_code, inputs=[input_code, language, max_chars, coalesce], outputs=output_code)
+    coalesce.change(fn=chunk_code, inputs=[input_code, language, max_chars, coalesce], outputs=output_code)
+    input_code.change(fn=chunk_code, inputs=[input_code, language, max_chars, coalesce], outputs=output_code)
 
 demo.launch()