Update app.py

app.py

@@ -54,14 +54,11 @@ def respond(
     if not message.strip():
         return history + [[message, "I'm sorry, I didn't receive any input. How can I help you today?"]]
     
-    # Log the incoming request
     logger.info(f"User {user_id} sent message - Length: {len(message)}")
     
     try:
-        # Prepare messages for the model
         messages = [{"role": "system", "content": system_message}]
         
-        # Format history correctly
         for user_msg, assistant_msg in history:
             if user_msg:
                 messages.append({"role": "user", "content": user_msg})
@@ -70,11 +67,12 @@ def respond(
         
         messages.append({"role": "user", "content": message})
         
-        # Generate response
         full_response = ""
         start_time = datetime.now()
         
-        # Stream the response
+        # Ensure proper response format for streaming
+        current_history = history.copy()
+        
         for message_chunk in client.chat_completion(
             messages, 
             max_tokens=max_tokens, 
@@ -85,13 +83,13 @@ def respond(
             token = message_chunk.choices[0].delta.content
             if token:
                 full_response += token
-                yield history + [[message, full_response]]
+                # Always ensure a list of lists with pairs
+                new_history = current_history + [[message, full_response]]
+                yield new_history
         
-        # Log completion
         time_taken = (datetime.now() - start_time).total_seconds()
         logger.info(f"Response generated for user {user_id} in {time_taken:.2f}s - Length: {len(full_response)}")
         
-        # Save conversation for audit/analytics
         conversation_data = {
             "timestamp": datetime.now().isoformat(),
             "user_id": user_id,
@@ -106,11 +104,10 @@ def respond(
         }
         save_conversation(user_id, conversation_data)
         
-        # Return final history with the complete response
-        # This is necessary if streaming didn't yield anything
-        if full_response:
-            return history + [[message, full_response]]
-        return history
+        # Ensure final response is properly formatted
+        if not full_response:
+            return history
+        return current_history + [[message, full_response]]
         
     except Exception as e:
         error_msg = f"An error occurred: {str(e)}"
@@ -334,8 +331,9 @@ with gr.Blocks(css=css, title=f"{COMPANY_NAME} AI Assistant") as demo:
             return f"<h3>Role: <span class='role-badge user-badge'>Standard User</span></h3>"
     
     def handle_role_permissions(role):
+        # Only disable settings for non-admin users
         if role == "admin":
-            return gr.update(visible=True)
+            return gr.update(visible=True, elem_classes=[])  # Remove any disabled classes
         else:
             return gr.update(visible=True, elem_classes=["setting-disabled"])
     
@@ -356,11 +354,12 @@ with gr.Blocks(css=css, title=f"{COMPANY_NAME} AI Assistant") as demo:
     
     # Chat functionality
     def chat_with_saved_params(message, history, uid, role):
-        # For non-admin users, use default values
+        # Get current values of parameters
         system_msg = system_message.value
-        tokens = max_tokens.value
-        temp = temperature.value
-        topp = top_p.value
+        tokens = max_tokens.value if role == "admin" else 512  # Default value for non-admin
+        temp = temperature.value if role == "admin" else 0.7   # Default value for non-admin
+        topp = top_p.value if role == "admin" else 0.95       # Default value for non-admin
+        
         return respond(message, history, system_msg, tokens, temp, topp, uid)
     
     msg_and_submit = msg.submit(
@@ -413,6 +412,11 @@ with gr.Blocks(css=css, title=f"{COMPANY_NAME} AI Assistant") as demo:
 with open("styles.css", "w") as f:
     f.write(css)
 
+css = css.replace(
+    ".setting-disabled {",
+    ".setting-disabled input, .setting-disabled button, .setting-disabled select, .setting-disabled textarea {"
+)
+
 if __name__ == "__main__":
     # Check if we're running in production
     if os.environ.get("PRODUCTION", "false").lower() == "true":