Update app.py

app.py

@@ -437,7 +437,7 @@ with gr.Blocks() as demo:
                 outputs=gr.Textbox(label="Fine-tuning Status", interactive=False),
                 title="⚙️ Fine-tuning (Optional)",
                 description="This section allows you to fine-tune the custom R1 model on a small subset of the ServiceNow dataset. This step is optional but can potentially improve the model's performance on ServiceNow-related tasks. **Note:** This process may take up to 5 minutes.",
-                submit_btn="🚀 Start Fine-tuning (QLoRA)" # Changed from submit_button_text to submit_btn
+                submit_btn="🚀 Start Fine-tuning (QLoRA)"
             ),
             gr.Interface(
                 fn=predict,
@@ -451,7 +451,7 @@ with gr.Blocks() as demo:
                 outputs=gr.Textbox(label="Custom R1 Output", lines=8, interactive=False),
                 title="✍️ Direct Generation",
                 description="Enter a prompt to generate text directly using the custom R1 model. This is standard text generation without retrieval augmentation.",
-                submit_btn="✨ Generate Text" # Changed from submit_button_text to submit_btn
+                submit_btn="✨ Generate Text"
             ),
             gr.Interface(
                 fn=compare_models,
@@ -468,27 +468,29 @@ with gr.Blocks() as demo:
                 ],
                 title="🆚 Model Comparison",
                 description="Enter a prompt to compare the text generation of your fine-tuned custom R1 model with the official DeepSeek-R1-Distill-Llama-8B model.",
-                submit_btn="⚖️ Compare Models" # Changed from submit_button_text to submit_btn
+                submit_btn="⚖️ Compare Models"
             ),
+            # Corrected gr.ChatInterface - inputs are NOT passed here
             gr.ChatInterface(
                 fn=chat_rag,
                 chatbot=gr.Chatbot(label="RAG Chatbot"),
                 textbox=gr.Textbox(placeholder="Ask a question to the RAG Chatbot...", lines=2, show_label=False),
-                inputs=[
-                    "textbox",
-                    gr.State([]), # chat_state
-                    gr.Slider(0.0, 1.5, step=0.1, value=0.7, label="Temperature"),
-                    gr.Slider(0.0, 1.0, step=0.05, value=0.9, label="Top-p"),
-                    gr.Slider(1, 2500, value=50, step=10, label="Min New Tokens"),
-                    gr.Slider(1, 2500, value=200, step=50, label="Max New Tokens")
-                ],
+                # inputs parameter REMOVED from gr.ChatInterface
                 title="💬 RAG Chat",
                 description="Chat with the custom R1 model, enhanced with retrieval-augmented memory. The model retrieves relevant info for informed responses.",
-                submit_button="➡️ Send",
-                clear_btn=None # Optional: You can add a clear button if needed
-            )
+                submit_button="➡️ Send", # Correct parameter name is submit_button for ChatInterface
+                clear_btn=None
+            ),
+             # Sliders defined OUTSIDE ChatInterface, but within the gr.Blocks() for the RAG Chat Tab
+            with gr.Row():
+                temperature_chat = gr.Slider(0.0, 1.5, step=0.1, value=0.7, label="Temperature", visible=False) # Initially hidden, can be shown if needed
+                top_p_chat = gr.Slider(0.0, 1.0, step=0.05, value=0.9, label="Top-p", visible=False) # Initially hidden, can be shown if needed
+                min_tokens_chat = gr.Slider(1, 2500, value=50, step=10, label="Min New Tokens", visible=False) # Initially hidden, can be shown if needed
+                max_tokens_chat = gr.Slider(1, 2500, value=200, step=50, label="Max New Tokens", visible=False) # Initially hidden, can be shown if needed
+
         ]
     ).render():
         pass # No need for extra elements outside the tabs now
 
+
 demo.launch()