Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@ logger = logging.getLogger("CompanyChatbot")
|
|
| 21 |
# Environment variables
|
| 22 |
HF_MODEL = os.environ.get("HF_MODEL", "mistralai/Mixtral-8x7B-Instruct-v0.1") # Smaller model
|
| 23 |
HF_API_TOKEN = os.environ.get("HF_API_TOKEN")
|
| 24 |
-
COMPANY_NAME = os.environ.get("COMPANY_NAME", "
|
| 25 |
DEFAULT_SYSTEM_PROMPT = os.environ.get("DEFAULT_SYSTEM_PROMPT",
|
| 26 |
f"You are {COMPANY_NAME}'s professional AI assistant. Be helpful, accurate, and concise.")
|
| 27 |
|
|
@@ -166,23 +166,102 @@ def login(username, password):
|
|
| 166 |
gr.update(visible=True, value="Invalid username or password")
|
| 167 |
)
|
| 168 |
|
| 169 |
-
|
| 170 |
css = """
|
| 171 |
-
|
| 172 |
-
|
| 173 |
-
|
| 174 |
-
|
| 175 |
-
|
| 176 |
-
|
| 177 |
-
.
|
| 178 |
-
|
| 179 |
-
|
| 180 |
-
|
| 181 |
-
|
| 182 |
-
.
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 186 |
"""
|
| 187 |
|
| 188 |
# Main application
|
|
|
|
| 21 |
# Environment variables
|
| 22 |
HF_MODEL = os.environ.get("HF_MODEL", "mistralai/Mixtral-8x7B-Instruct-v0.1") # Smaller model
|
| 23 |
HF_API_TOKEN = os.environ.get("HF_API_TOKEN")
|
| 24 |
+
COMPANY_NAME = os.environ.get("COMPANY_NAME", "RS")
|
| 25 |
DEFAULT_SYSTEM_PROMPT = os.environ.get("DEFAULT_SYSTEM_PROMPT",
|
| 26 |
f"You are {COMPANY_NAME}'s professional AI assistant. Be helpful, accurate, and concise.")
|
| 27 |
|
|
|
|
| 166 |
gr.update(visible=True, value="Invalid username or password")
|
| 167 |
)
|
| 168 |
|
| 169 |
+
|
| 170 |
css = """
|
| 171 |
+
|
| 172 |
+
body {
|
| 173 |
+
font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
|
| 174 |
+
background-color: #f9f9f9;
|
| 175 |
+
}
|
| 176 |
+
|
| 177 |
+
.container {
|
| 178 |
+
max-width: 1400px !important;
|
| 179 |
+
margin: auto;
|
| 180 |
+
}
|
| 181 |
+
|
| 182 |
+
.setting-panel {
|
| 183 |
+
background-color: #f0f4f8;
|
| 184 |
+
border-radius: 10px;
|
| 185 |
+
padding: 15px;
|
| 186 |
+
box-shadow: 0 2px 6px rgba(0,0,0,0.1);
|
| 187 |
+
}
|
| 188 |
+
|
| 189 |
+
.chat-container {
|
| 190 |
+
border-radius: 10px;
|
| 191 |
+
box-shadow: 0 2px 6px rgba(0,0,0,0.1);
|
| 192 |
+
background-color: white;
|
| 193 |
+
}
|
| 194 |
+
|
| 195 |
+
.company-header {
|
| 196 |
+
background-color: #2c3e50;
|
| 197 |
+
color: white;
|
| 198 |
+
padding: 15px;
|
| 199 |
+
border-radius: 10px 10px 0 0;
|
| 200 |
+
margin-bottom: 15px;
|
| 201 |
+
}
|
| 202 |
+
|
| 203 |
+
.footer {
|
| 204 |
+
text-align: center;
|
| 205 |
+
margin-top: 20px;
|
| 206 |
+
color: #666;
|
| 207 |
+
font-size: 0.8em;
|
| 208 |
+
}
|
| 209 |
+
|
| 210 |
+
.message-user {
|
| 211 |
+
background-color: #e6f7ff !important;
|
| 212 |
+
border-radius: 15px 15px 0 15px !important;
|
| 213 |
+
}
|
| 214 |
+
|
| 215 |
+
.message-bot {
|
| 216 |
+
background-color: #f0f0f0 !important;
|
| 217 |
+
border-radius: 15px 15px 15px 0 !important;
|
| 218 |
+
}
|
| 219 |
+
|
| 220 |
+
.login-container {
|
| 221 |
+
max-width: 500px;
|
| 222 |
+
margin: 50px auto;
|
| 223 |
+
padding: 30px;
|
| 224 |
+
background-color: white;
|
| 225 |
+
border-radius: 10px;
|
| 226 |
+
box-shadow: 0 4px 10px rgba(0,0,0,0.1);
|
| 227 |
+
}
|
| 228 |
+
|
| 229 |
+
.login-header {
|
| 230 |
+
text-align: center;
|
| 231 |
+
margin-bottom: 30px;
|
| 232 |
+
}
|
| 233 |
+
|
| 234 |
+
.error-message {
|
| 235 |
+
color: #e74c3c;
|
| 236 |
+
background-color: #fdedeb;
|
| 237 |
+
padding: 10px;
|
| 238 |
+
border-radius: 5px;
|
| 239 |
+
margin-bottom: 15px;
|
| 240 |
+
font-size: 14px;
|
| 241 |
+
}
|
| 242 |
+
|
| 243 |
+
.role-badge {
|
| 244 |
+
font-size: 12px;
|
| 245 |
+
padding: 3px 8px;
|
| 246 |
+
border-radius: 10px;
|
| 247 |
+
margin-left: 10px;
|
| 248 |
+
}
|
| 249 |
+
|
| 250 |
+
.admin-badge {
|
| 251 |
+
background-color: #e74c3c;
|
| 252 |
+
color: white;
|
| 253 |
+
}
|
| 254 |
+
|
| 255 |
+
.user-badge {
|
| 256 |
+
background-color: #3498db;
|
| 257 |
+
color: white;
|
| 258 |
+
}
|
| 259 |
+
|
| 260 |
+
.setting-disabled {
|
| 261 |
+
opacity: 0.5;
|
| 262 |
+
pointer-events: none;
|
| 263 |
+
}
|
| 264 |
+
|
| 265 |
"""
|
| 266 |
|
| 267 |
# Main application
|