:root {
  --primary-color: #007bff;
  --secondary-color: #6c757d;
  --light-bg: #f8f9fa;
  --dark-text: #343a40;
  --light-text: #ffffff;
  --user-message-bg: #e7f5ff;
  --assistant-message-bg: #f1f3f5;
  --error-bg: #f8d7da;
  --error-text: #721c24;
  --error-border: #f5c6cb;
  --success-color: #28a745;
  --danger-color: #dc3545;
  --border-color: #dee2e6;
  --input-bg: #fff;
  --body-bg: #e9ecef; /* Lighter grey background */
}

body {
  font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica,
    Arial, sans-serif, "Apple Color Emoji", "Segoe UI Emoji", "Segoe UI Symbol";
  max-width: 800px;
  margin: 20px auto; /* Add top/bottom margin */
  padding: 20px;
  background-color: var(--body-bg);
  color: var(--dark-text);
  display: flex; /* Use flexbox for layout */
  flex-direction: column;
  height: calc(100vh - 40px); /* Full viewport height minus body margin */
  box-sizing: border-box;
}

h1 {
  text-align: center;
  color: var(--primary-color);
  margin-bottom: 20px;
}

#model-select-container {
  /* Wrapper for select */
  margin-bottom: 15px;
  display: flex;
  align-items: center;
  gap: 10px;
  padding: 10px;
  background-color: var(--light-bg);
  border-radius: 8px;
  border: 1px solid var(--border-color);
}

.mode-switch {
  display: flex;
  justify-content: center;
  gap: 8px;
  margin: 12px 0 16px 0;
}

.mode-btn {
  background-color: var(--light-bg);
  color: var(--dark-text);
  border: 1px solid var(--border-color);
}

.mode-btn.active {
  background-color: var(--primary-color);
  color: var(--light-text);
}

#model-select-container label {
  font-weight: bold;
  color: var(--secondary-color);
}

#model-select {
  flex-grow: 1;
  padding: 8px 12px;
  border: 1px solid var(--border-color);
  border-radius: 4px;
  background-color: var(--input-bg);
  cursor: pointer;
}

/* style provider dropdown like model dropdown */
#provider-select {
  flex-grow: 1;
  padding: 8px 12px;
  border: 1px solid var(--border-color);
  border-radius: 4px;
  background-color: var(--input-bg);
  cursor: pointer;
}

#chat-container {
  border: 1px solid var(--border-color);
  padding: 15px;
  height: 100%; /* Take remaining height */
  overflow-y: auto;
  margin-bottom: 15px;
  background-color: var(--input-bg); /* White background for chat */
  border-radius: 8px;
  flex-grow: 1; /* Allow chat to grow */
  display: flex;
  flex-direction: column;
  gap: 10px; /* Space between messages */
}

#input-area {
  /* Wrapper for input and buttons */
  display: flex;
  gap: 10px;
  align-items: center;
  padding: 10px;
  background-color: var(--light-bg);
  border-radius: 8px;
  border: 1px solid var(--border-color);
}

#image-controls {
  display: flex;
  flex-direction: column;
  gap: 10px;
  padding: 10px;
}

.image-row2 {
  display: flex;
  gap: 10px;
  align-items: center;
  flex-wrap: wrap;
}

#image-prompt {
  width: 100%;
  min-height: 140px;
  padding: 10px 12px;
  border: 1px solid var(--border-color);
  border-radius: 8px;
  background-color: var(--input-bg);
  box-sizing: border-box;
  resize: vertical;
}

.panel-image {
  margin-top: 16px;
  padding: 16px;
  background-color: var(--input-bg);
  border: 1px solid var(--border-color);
  border-radius: 10px;
  box-shadow: 0 2px 12px rgba(0, 0, 0, 0.06);
  border-left: 4px solid var(--success-color);
}

.panel-image h3 {
  margin: 0 0 8px 0;
  color: var(--success-color);
}

#image-status {
  margin-top: 8px;
  color: var(--secondary-color);
  font-size: 0.95em;
}

.image-actions {
  display: flex;
  justify-content: flex-end;
  align-items: center;
  gap: 8px;
  margin-bottom: 6px;
}

.download-btn {
  display: inline-flex;
  align-items: center;
  justify-content: center;
  width: 28px;
  height: 28px;
  border-radius: 50%;
  background-color: var(--light-bg);
  color: var(--dark-text);
  border: 1px solid var(--border-color);
  text-decoration: none;
  font-size: 16px;
}
.download-btn:hover {
  background-color: var(--primary-color);
  color: var(--light-text);
}

.spinner {
  width: 18px;
  height: 18px;
  border: 2px solid var(--border-color);
  border-top-color: var(--success-color);
  border-radius: 50%;
  animation: spin 0.8s linear infinite;
}

@keyframes spin {
  to {
    transform: rotate(360deg);
  }
}

#message-input {
  flex-grow: 1; /* Take available space */
  padding: 10px 15px;
  border: 1px solid var(--border-color);
  border-radius: 20px; /* Rounded input */
  margin-bottom: 0; /* Remove default margin */
  resize: none; /* Prevent textarea resize */
  height: 40px; /* Fixed height */
  box-sizing: border-box;
}

button {
  padding: 10px 20px;
  border: none;
  border-radius: 20px; /* Rounded buttons */
  cursor: pointer;
  font-weight: bold;
  transition: background-color 0.2s ease, transform 0.1s ease;
  white-space: nowrap; /* Prevent button text wrapping */
}

button:hover {
  opacity: 0.9;
  transform: scale(1.02); /* Slight scale effect */
}

button:active {
  transform: scale(0.98); /* Press effect */
}

#send-button {
  background-color: var(--primary-color);
  color: var(--light-text);
}
#send-button:hover {
  background-color: #0056b3; /* Darker blue */
}

#stop-button {
  background-color: var(--danger-color);
  color: var(--light-text);
  display: none; /* Hidden by default */
}
#stop-button:hover {
  background-color: #c82333; /* Darker red */
}

#stop-image {
  background-color: var(--danger-color);
  color: var(--light-text);
  display: none;
}
#stop-image:hover {
  background-color: #c82333;
}

#predefined-button {
  background-color: var(--secondary-color);
  color: var(--light-text);
}
#predefined-button:hover {
  background-color: #5a6268; /* Darker grey */
}

button:disabled {
  background-color: #adb5bd;
  cursor: not-allowed;
  transform: none;
}

.message {
  padding: 10px 15px;
  border-radius: 15px; /* Rounded message bubbles */
  max-width: 80%; /* Limit message width */
  word-wrap: break-word; /* Wrap long words */
  line-height: 1.4;
}

.user-message {
  background-color: var(--user-message-bg);
  color: var(--dark-text);
  align-self: flex-end; /* Align user messages to the right */
  border-bottom-right-radius: 5px; /* Slightly different corner */
  margin-left: auto; /* Push to right */
}

.assistant-message {
  background-color: var(--assistant-message-bg);
  color: var(--dark-text);
  align-self: flex-start; /* Align assistant messages to the left */
  border-bottom-left-radius: 5px; /* Slightly different corner */
  margin-right: auto; /* Push to left */
}

.system-message {
  background-color: transparent; /* No background */
  color: var(--secondary-color);
  font-style: italic;
  text-align: center;
  font-size: 0.9em;
  align-self: center; /* Center system messages */
  max-width: 100%;
  padding: 5px 10px;
}

.error-message {
  background-color: var(--error-bg);
  color: var(--error-text);
  border: 1px solid var(--error-border);
  align-self: stretch; /* Stretch error messages */
  text-align: center;
  max-width: 100%;
}

#predefined-messages {
  margin-top: 15px;
  padding: 15px;
  background-color: var(--light-bg);
  border: 1px solid var(--border-color);
  border-radius: 8px;
}
#predefined-messages h3 {
  margin-top: 0;
  margin-bottom: 10px;
  color: var(--secondary-color);
  text-align: center;
}
#predefined-messages button {
  background-color: var(--input-bg);
  color: var(--primary-color);
  border: 1px solid var(--primary-color);
  margin: 5px;
  padding: 8px 15px;
  border-radius: 15px;
}
#predefined-messages button:hover {
  background-color: var(--primary-color);
  color: var(--light-text);
}
#predefined-messages button:last-child {
  /* Style the close button differently */
  background-color: var(--secondary-color);
  color: var(--light-text);
  border: 1px solid var(--secondary-color);
}
#predefined-messages button:last-child:hover {
  background-color: #5a6268;
}

/* Custom Scrollbar (optional, WebKit browsers) */
#chat-container::-webkit-scrollbar {
  width: 8px;
}
#chat-container::-webkit-scrollbar-track {
  background: var(--light-bg);
  border-radius: 4px;
}
#chat-container::-webkit-scrollbar-thumb {
  background-color: var(--secondary-color);
  border-radius: 4px;
  border: 2px solid var(--light-bg);
}
#chat-container::-webkit-scrollbar-thumb:hover {
  background-color: #5a6268;
}