Update app.py

app.py

@@ -6,7 +6,7 @@ from groq import Groq
 api_key = os.environ.get("GROQ_API_KEY")
 client = Groq(api_key=api_key)
 
-# 2. UI Styling for Minimalist Full-Screen
+# 2. UI Styling
 custom_css = """
 .gradio-container { 
     max-width: 100% !important; 
@@ -22,7 +22,6 @@ custom_css = """
      background: white; 
 }
 
-/* Scroll fixes for inputs */
 #system-prompt-container textarea, .scroll-input textarea {
     height: 45px !important;
     max-height: 45px !important;
@@ -30,7 +29,6 @@ custom_css = """
     resize: none !important;
 }
 
-/* FIXED ICON BUTTON STYLING */
 #settings-btn {
     width: 44px !important;
     height: 44px !important;
@@ -80,12 +78,13 @@ def chat_with_groq(message, history, model, temperature, system_prompt):
         yield f"⚠️ API Error: {str(e)}"
 
 # 4. Interface Setup
-with gr.Blocks(css=custom_css) as demo:
+with gr.Blocks() as demo:
     sidebar_state = gr.State(False)
 
-    # Header with minimalist icon button
+    # Header Section - Fixed scale issues
     with gr.Row():
-        gr.Markdown("# 🌿🤖 SageBot", scale=20)
+        with gr.Column(scale=20):
+            gr.Markdown("# 🌿🤖 SageBot")
         with gr.Column(scale=1, min_width=50):
             settings_btn = gr.Button("⚙️", elem_id="settings-btn")
     
@@ -104,7 +103,7 @@ with gr.Blocks(css=custom_css) as demo:
                 )
                 submit_btn = gr.Button("Send", variant="primary", scale=1, min_width=80)
         
-        # Sidebar Area (Hidden by default)
+        # Sidebar Area
         with gr.Column(scale=1, visible=False, elem_classes="sidebar-panel") as sidebar_col:
             gr.Markdown("### ⚙️ Global Settings")
             model_choice = gr.Dropdown(
@@ -130,7 +129,7 @@ with gr.Blocks(css=custom_css) as demo:
             </div>
             """)
 
-    # --- Sidebar Toggle Logic ---
+    # Sidebar Toggle
     def toggle_sidebar(current_state):
         new_state = not current_state
         return gr.update(visible=new_state), new_state
@@ -159,5 +158,6 @@ with gr.Blocks(css=custom_css) as demo:
     )
     clear.click(lambda: [], None, chatbot, queue=False)
 
+# Pass CSS into launch() instead of Blocks
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(css=custom_css)