Update app.py

app.py

@@ -24,6 +24,8 @@ def respond(
     temperature: float,
     top_p: float,
 ):
+    """Handles chatbot responses with Perplexity AI."""
+    
     if model_choice not in AVAILABLE_MODELS:
         return "Error: Invalid model selection."
 
@@ -68,6 +70,7 @@ def respond(
         yield f"Error: {str(e)}"
 
 def check_password(input_password):
+    """Validates the password before showing the chat interface."""
     if input_password == PASSWORD:
         return gr.update(visible=False), gr.update(visible=True)
     else:
@@ -87,31 +90,27 @@ with gr.Blocks() as demo:
         system_prompt = gr.Textbox(
             value="You are a helpful assistant.", label="System message"
         )
+        model_choice = gr.Dropdown(
+            choices=list(AVAILABLE_MODELS.keys()),
+            value=list(AVAILABLE_MODELS.keys())[0],
+            label="Select Model"
+        )
+        max_tokens = gr.Slider(
+            minimum=1, maximum=30000, value=2048, step=100, label="Max new tokens"
+        )
+        temperature = gr.Slider(
+            minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"
+        )
+        top_p = gr.Slider(
+            minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"
+        )
+
         chat = gr.ChatInterface(
             respond,
             chatbot=gr.Chatbot(height=400),  # Set the desired height here
-            additional_inputs=[system_prompt],  # Include system message explicitly
+            additional_inputs=[system_prompt, model_choice, max_tokens, temperature, top_p]  # Pass extra parameters
         )
 
-        with gr.Column():
-            model_choice = gr.Dropdown(
-                choices=list(AVAILABLE_MODELS.keys()),
-                value=list(AVAILABLE_MODELS.keys())[0],
-                label="Select Model"
-            )
-            max_tokens = gr.Slider(
-                minimum=1, maximum=30000, value=2048, step=100, label="Max new tokens"
-            )
-            temperature = gr.Slider(
-                minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"
-            )
-            top_p = gr.Slider(
-                minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"
-            )
-
-        # Update chat interface to include additional inputs
-        chat.additional_inputs.extend([model_choice, max_tokens, temperature, top_p])
-
     submit_button.click(
         check_password, inputs=password_input, outputs=[password_input, chat_interface]
     )