Update app.py

app.py

@@ -33,7 +33,7 @@ def respond(
 
     response = ""
     citations = []
-    
+
     stream = px_client.chat.completions.create(
         model=AVAILABLE_MODELS[model_choice],
         messages=messages,
@@ -42,7 +42,7 @@ def respond(
         top_p=top_p,
         stream=True,
     )
-    
+
     for chunk in stream:
         if "choices" in chunk:
             token = chunk.choices[0].delta.content or ""
@@ -50,7 +50,7 @@ def respond(
             yield response  # Stream response as it arrives
         if "citations" in chunk:
             citations = chunk["citations"]
-    
+
     # Append citations as clickable links
     if citations:
         citation_text = "\n\nSources:\n" + "\n".join(
@@ -65,7 +65,14 @@ def check_password(input_password):
     else:
         return gr.update(value="", interactive=True), gr.update(visible=False)
 
-with gr.Blocks() as demo:
+# Custom CSS to adjust the chat interface height
+custom_css = """
+#chat_interface .overflow-y-auto {
+    height: 600px;
+}
+"""
+
+with gr.Blocks(css=custom_css) as demo:
     with gr.Column():
         password_input = gr.Textbox(
             type="password", label="Enter Password", interactive=True
@@ -74,7 +81,7 @@ with gr.Blocks() as demo:
         error_message = gr.Textbox(
             label="Error", visible=False, interactive=False
         )
-    
+
     with gr.Column(visible=False) as chat_interface:
         system_prompt = gr.Textbox(
             value="You are a helpful assistant.", label="System message"
@@ -82,9 +89,9 @@ with gr.Blocks() as demo:
         chat = gr.ChatInterface(
             respond,
             additional_inputs=[],
-            height=600  # Increased height for better visibility
+            elem_id="chat_interface"  # Assign an element ID for CSS targeting
         )
-        
+
         with gr.Column():
             model_choice = gr.Dropdown(
                 choices=list(AVAILABLE_MODELS.keys()),
@@ -100,10 +107,10 @@ with gr.Blocks() as demo:
             top_p = gr.Slider(
                 minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"
             )
-    
+
     submit_button.click(
         check_password, inputs=password_input, outputs=[password_input, chat_interface]
     )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)