Update app.py

app.py

@@ -1,9 +1,8 @@
 import gradio as gr
 
 # Placeholder for model loading (adjust as needed for your specific models)
-# In a real application, these would load actual models
 def load_model(model_name):
-    print(f"Loading {model_name}...")  # Indicate model loading
+    print(f"Loading {model_name}...")
     # Simulate different model behaviors (replace with actual model logic)
     if model_name == "DeepSeek-R1-Distill-Qwen-32B":
         return lambda input_text, history: f"Distilled Model Response to: {input_text}"
@@ -34,44 +33,42 @@ def chatbot(input_text, history, model_choice, system_message, max_new_tokens, t
     else:
         model_function = lambda x, h: "Please select a model."
 
-    # Simulate model response with parameters (replace with actual model inference)
-    # In a real application, you would pass these parameters to your model
+    # Simulate model response with parameters
     response = model_function(input_text, history)
-    response = f"**System Message:** {system_message}\n\n**Model Response:** {response}\n\n" \
-               f"**Parameters Used:**\n- Max New Tokens: {max_new_tokens}\n- Temperature: {temperature}\n- Top-p: {top_p}"
-
-    history.append((input_text, response))
-    return history, history, ""  # Update both chatbot output and state
+    # Format the response for display (without parameter details in the main chat)
+    display_response = f"{response}"
 
+    history.append((input_text, display_response))
+    return history, history, "", model_choice, system_message, max_new_tokens, temperature, top_p  # Clear input, keep other parameters
 
 # --- Gradio Interface ---
-with gr.Blocks(theme=gr.themes.Soft()) as demo:  # Apply a theme
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown(
         """
         # DeepSeek Chatbot
         Created by [ruslanmv.com](https://ruslanmv.com/)
-        
+
         This is a demo of different DeepSeek models. Select a model, type your message, and click "Submit".
         You can also adjust optional parameters like system message, max new tokens, temperature, and top-p.
         """
     )
 
     with gr.Row():
-        with gr.Column(scale=4): # Make chatbot take more space
+        with gr.Column():
             chatbot_output = gr.Chatbot(label="DeepSeek Chatbot", height=500)
             msg = gr.Textbox(label="Your Message", placeholder="Type your message here...")
-
             with gr.Row():
                 submit_btn = gr.Button("Submit", variant="primary")
                 clear_btn = gr.ClearButton([msg, chatbot_output])
-        
-        with gr.Column(scale=1):
+
+    # Options moved below the chat interface
+    with gr.Row():
+        with gr.Accordion("Options", open=True): # Changed label to "Options"
             model_choice = gr.Radio(
                 choices=["DeepSeek-R1-Distill-Qwen-32B", "DeepSeek-R1", "DeepSeek-R1-Zero"],
                 label="Choose a Model",
-                value="DeepSeek-R1"  # Default model
+                value="DeepSeek-R1"
             )
-
             with gr.Accordion("Optional Parameters", open=False):
                 system_message = gr.Textbox(
                     label="System Message",
@@ -87,6 +84,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:  # Apply a theme
                 top_p = gr.Slider(
                     minimum=0.10, maximum=1.00, value=0.90, label="Top-p (nucleus sampling)"
                 )
+    
 
     # Maintain chat history
     chat_history = gr.State([])
@@ -95,12 +93,12 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:  # Apply a theme
     submit_btn.click(
         chatbot,
         [msg, chat_history, model_choice, system_message, max_new_tokens, temperature, top_p],
-        [chatbot_output, chat_history, msg],
+        [chatbot_output, chat_history, msg, model_choice, system_message, max_new_tokens, temperature, top_p],
     )
     msg.submit(
         chatbot,
         [msg, chat_history, model_choice, system_message, max_new_tokens, temperature, top_p],
-        [chatbot_output, chat_history, msg],
+        [chatbot_output, chat_history, msg, model_choice, system_message, max_new_tokens, temperature, top_p],
     )
 
 # Launch the demo