add deepseek-r1

app.py

@@ -6,6 +6,11 @@ from typing import Dict, List, Optional, Tuple
 
 import gradio as gr
 from loguru import logger
+try:
+  from utils import init_logger
+  logger = init_logger()
+except Exception as e:
+    logger.info(f"import_error: {str(e)}")
 
 import sambanova_gradio
 from sambanova_gradio import get_fn
@@ -311,6 +316,23 @@ with gr.Blocks(css_paths="app.css", title="Qwen2.5 Coder Artifacts (+SambaNova+F
 
                         input = antd.InputTextarea(size="middle", allow_clear=True, placeholder="Please enter what kind of application you want")
                         code_language = gr.Textbox(label="code language", lines=1)
+                        LLM_model = gr.Dropdown(
+                            choices=[
+                                "Qwen2.5-Coder-32B-Instruct",
+                                "DeepSeek-R1-Distill-Llama-70B",
+                                "Llama-3.1-Tulu-3-405B",
+                                "Meta-Llama-3.3-70B-Instruct",
+                                "Meta-Llama-3.2-3B-Instruct",
+                                "Meta-Llama-3.2-1B-Instruct",
+                                "Meta-Llama-3.1-70B-Instruct", 
+                                "Meta-Llama-3.1-405B-Instruct", 
+                                "Meta-Llama-3.1-8B-Instruct",
+                                "QwQ-32B-Preview",
+                                "Qwen2.5-72B-Instruct",
+                                ],
+                            label="LLM model",
+                            value="Qwen2.5-Coder-32B-Instruct",
+                        )                         
                         btn = antd.Button("send", type="primary", size="large")                        
 
                         with antd.Flex(gap="small", wrap=True):
@@ -359,10 +381,10 @@ with gr.Blocks(css_paths="app.css", title="Qwen2.5 Coder Artifacts (+SambaNova+F
                                 sandbox = gr.HTML(elem_classes="html_content")
                             with antd.Tabs.Item(key="render_other", label="Render_Code") as render_other:
                                 sandbox_other = legacy.Markdown()
-                    with antd.Row():
-                        code_syntax_check = gr.Textbox(label="check code syntax: ", lines=1, interactive=False)
+                    code_syntax_check = gr.Textbox(label="check code syntax: ", lines=1, interactive=False)
+                    think_content = gr.Textbox(label="think_content: ", lines=10, interactive=False, visible=False)
 
-        def generation_code(query: Optional[str], code_language, _setting: Dict[str, str], _history: Optional[History]):
+        def generation_code(query: Optional[str], code_language, _setting: Dict[str, str], _history: Optional[History], LLM_model):
             if query is None:
                 query = ''
             if _history is None:
@@ -387,7 +409,7 @@ with gr.Blocks(css_paths="app.css", title="Qwen2.5 Coder Artifacts (+SambaNova+F
 
             # Get the model from sambanova_gradio
             fn = get_fn(
-                model_name='Qwen2.5-Coder-32B-Instruct',
+                model_name=LLM_model,
                 preprocess=preprocess,
                 postprocess=postprocess,
                 api_key=YOUR_API_TOKEN
@@ -407,6 +429,14 @@ with gr.Blocks(css_paths="app.css", title="Qwen2.5 Coder Artifacts (+SambaNova+F
                 }
 
             assistant_response = response_text
+            pattern = r'<think>\n(.*?)</think>\n'
+            match = re.search(pattern, assistant_response, flags=re.DOTALL)
+            if match:
+                think_content_val = match.group(1)
+            else:
+                think_content_val = ""
+            assistant_response = re.sub(pattern, '', assistant_response, flags=re.DOTALL)
+
             local_history.append([query, assistant_response])
 
             code = remove_code_block(assistant_response, language=code_language)
@@ -435,6 +465,8 @@ with gr.Blocks(css_paths="app.css", title="Qwen2.5 Coder Artifacts (+SambaNova+F
                 render_other: render_other_val,
                 sandbox_other: sandbox_other_val,
                 code_syntax_check: code_syntax_check_val,
+                think_content: think_content_val,
+                think_content: gr.update(visible=(think_content_val!="")),
             }
 
         def code_language_example_change(code_language):
@@ -453,9 +485,9 @@ with gr.Blocks(css_paths="app.css", title="Qwen2.5 Coder Artifacts (+SambaNova+F
             )
         
         btn.click(generation_code,
-                  inputs=[input, code_language, setting, history_state],
+                  inputs=[input, code_language, setting, history_state, LLM_model],
                   outputs=[code_output, history_state, sandbox, state_tab, code_drawer,
-                           render_html, render_other, sandbox_other, code_syntax_check])
+                           render_html, render_other, sandbox_other, code_syntax_check, think_content, think_content])
 
         clear_btn.click(clear_history, inputs=[], outputs=[history_state])