Update app.py

app.py

@@ -8,10 +8,6 @@ HF_TOKEN = os.getenv("HF_TOKEN")  # Replace with your actual token if not using
 # Initialize the Gradio Client for the specified API
 client = Client("on1onmangoes/CNIHUB10724v9", hf_token=HF_TOKEN)
 
-# Authentication function
-def login(username, password):
-    return username == "your_username" and password == "your_password"  # Update with actual credentials
-
 # Function to handle chat API call
 def stream_chat_with_rag(message, client_name, system_prompt, num_retrieved_docs, num_docs_final, temperature, max_new_tokens, top_p, top_k, penalty):
     response = client.predict(
@@ -54,95 +50,81 @@ with gr.Blocks() as app:
         password_input = gr.Textbox(label="Password", placeholder="Enter your password", type="password")
     
     with gr.Tab("Chat"):
-        chatbot = gr.Chatbot(fill_height=True)  # Create a chatbot interface
-        
-        # Create an accordion for additional input parameters
-        additional_inputs_accordion = gr.Accordion(label="⚙️ Parameters", open=False, render=False)
-        
-        with additional_inputs_accordion:
-            client_name_dropdown = gr.Dropdown(
-                label="Select Client",
-                choices=["rosariarossi", "bianchifiordaliso", "lorenzoverdi"],
-                value="rosariarossi",
-                render=False
-            )
-            system_prompt_input = gr.Textbox(
-                value="You are an expert assistant",
-                label="System Prompt",
-                render=False
-            )
-            num_retrieved_docs_slider = gr.Slider(
-                label="Number of Initial Documents to Retrieve",
-                minimum=1,
-                maximum=10,
-                step=1,
-                value=10,
-                render=False
-            )
-            num_docs_final_slider = gr.Slider(
-                label="Number of Final Documents to Retrieve",
-                minimum=1,
-                maximum=10,
-                step=1,
-                value=9,
-                render=False
-            )
-            temperature_slider = gr.Slider(
-                label="Temperature",
-                minimum=0.2,
-                maximum=1,
-                step=0.1,
-                value=0,
-                render=False
-            )
-            max_new_tokens_slider = gr.Slider(
-                label="Max New Tokens",
-                minimum=128,
-                maximum=8192,
-                step=1,
-                value=1024,
-                render=False
-            )
-            top_p_slider = gr.Slider(
-                label="Top P",
-                minimum=0.0,
-                maximum=1.0,
-                step=0.1,
-                value=1.0,
-                render=False
-            )
-            top_k_slider = gr.Slider(
-                label="Top K",
-                minimum=1,
-                maximum=20,
-                step=1,
-                value=20,
-                render=False
-            )
-            penalty_slider = gr.Slider(
-                label="Repetition Penalty",
-                minimum=0.0,
-                maximum=2.0,
-                step=0.1,
-                value=1.2,
-                render=False
-            )
-        
+        chatbot = gr.Chatbot()  # Create a chatbot interface
+
         chat_interface = gr.ChatInterface(
             fn=stream_chat_with_rag,
             chatbot=chatbot,
             fill_height=True,
-            additional_inputs_accordion=additional_inputs_accordion,
+            additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
             additional_inputs=[
-                client_name_dropdown,
-                system_prompt_input,
-                num_retrieved_docs_slider,
-                num_docs_final_slider,
-                temperature_slider,
-                max_new_tokens_slider,
-                top_p_slider,
-                top_k_slider,
-                penalty_slider,
+                gr.Dropdown(
+                    ['rosariarossi', 'bianchifiordaliso', 'lorenzoverdi'],
+                    value="rosariarossi",
+                    label="Select Client",
+                    render=False,
+                ),
+                gr.Textbox(
+                    value="You are an expert assistant",
+                    label="System Prompt",
+                    render=False,
+                ),
+                gr.Slider(
+                    minimum=1,
+                    maximum=10,
+                    step=1,
+                    value=10,
+                    label="Number of Initial Documents to Retrieve",
+                    render=False,
+                ),
+                gr.Slider(
+                    minimum=1,
+                    maximum=10,
+                    step=1,
+                    value=9,
+                    label="Number of Final Documents to Retrieve",
+                    render=False,
+                ),
+                gr.Slider(
+                    minimum=0.2,
+                    maximum=1,
+                    step=0.1,
+                    value=0,
+                    label="Temperature",
+                    render=False,
+                ),
+                gr.Slider(
+                    minimum=128,
+                    maximum=8192,
+                    step=1,
+                    value=1024,
+                    label="Max new tokens",
+                    render=False,
+                ),
+                gr.Slider(
+                    minimum=0.0,
+                    maximum=1.0,
+                    step=0.1,
+                    value=1.0,
+                    label="Top P",
+                    render=False,
+                ),
+                gr.Slider(
+                    minimum=1,
+                    maximum=20,
+                    step=1,
+                    value=20,
+                    label="Top K",
+                    render=False,
+                ),
+                gr.Slider(
+                    minimum=0.0,
+                    maximum=2.0,
+                    step=0.1,
+                    value=1.2,
+                    label="Repetition Penalty",
+                    render=False,
+                ),
             ],
         )