Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -23,7 +23,6 @@ try:
|
|
| 23 |
except Exception:
|
| 24 |
requests = None
|
| 25 |
|
| 26 |
-
|
| 27 |
# =====================================================
|
| 28 |
# CORE CONFIG
|
| 29 |
# =====================================================
|
|
@@ -39,7 +38,6 @@ LOG_DIR = "logs"
|
|
| 39 |
os.makedirs(LOG_DIR, exist_ok=True)
|
| 40 |
LOG_PATH = os.path.join(LOG_DIR, "conversations.jsonl")
|
| 41 |
|
| 42 |
-
|
| 43 |
# =====================================================
|
| 44 |
# KNOWLEDGE BASE (CourseContent / CaseStudyContent)
|
| 45 |
# =====================================================
|
|
@@ -56,7 +54,6 @@ CASE_PREFIX = "CaseStudyContent__"
|
|
| 56 |
def _is_placeholder_name(name: str) -> bool:
|
| 57 |
return "placeholder" in (name or "").lower()
|
| 58 |
|
| 59 |
-
|
| 60 |
# =====================================================
|
| 61 |
# MODES (single customization surface) ****** CUSTOMIZE ******
|
| 62 |
# =====================================================
|
|
@@ -185,7 +182,6 @@ Matching how the persona described in the source would be in a realistic busines
|
|
| 185 |
STUDENT_MODES = list(MODE_REGISTRY.keys())
|
| 186 |
DEFAULT_MODE = STUDENT_MODES[0] if STUDENT_MODES else "Course Detailer"
|
| 187 |
|
| 188 |
-
|
| 189 |
# =====================================================
|
| 190 |
# OPTIONAL WEB SEARCH (OFF by default) ****** CUSTOMIZE ******
|
| 191 |
# =====================================================
|
|
@@ -199,7 +195,6 @@ def web_search_snippets(query: str, max_chars: int = 1500) -> str:
|
|
| 199 |
return ""
|
| 200 |
return ""
|
| 201 |
|
| 202 |
-
|
| 203 |
# =====================================================
|
| 204 |
# MEMORY + LOGGING
|
| 205 |
# =====================================================
|
|
@@ -222,7 +217,6 @@ def log_turn(mode: str, user_msg: str, assistant_msg: str, upload_meta=None) ->
|
|
| 222 |
with open(LOG_PATH, "a", encoding="utf-8") as f:
|
| 223 |
f.write(json.dumps(rec, ensure_ascii=False) + "\n")
|
| 224 |
|
| 225 |
-
|
| 226 |
# =====================================================
|
| 227 |
# OUTPUT CLEANING
|
| 228 |
# =====================================================
|
|
@@ -235,7 +229,6 @@ def strip_think(text: str) -> str:
|
|
| 235 |
cleaned = _THINK_RE.sub("", text).strip()
|
| 236 |
return cleaned if cleaned else "(No visible answer returned. Please re-try.)"
|
| 237 |
|
| 238 |
-
|
| 239 |
# =====================================================
|
| 240 |
# FILE EXTRACTION
|
| 241 |
# =====================================================
|
|
@@ -412,7 +405,6 @@ def files_to_text(upload_paths) -> tuple[str, list[dict], str]:
|
|
| 412 |
badge = f"Attached: {', '.join(names[:MAX_FILES])}" if names else ""
|
| 413 |
return combined, meta_list, badge
|
| 414 |
|
| 415 |
-
|
| 416 |
# =====================================================
|
| 417 |
# STATIC KB INGESTION
|
| 418 |
# =====================================================
|
|
@@ -470,7 +462,6 @@ COURSE_PATHS, CASE_PATHS = _list_kb_files()
|
|
| 470 |
COURSE_KB_TEXT, COURSE_PLACEHOLDER_ONLY = _kb_extract(COURSE_PATHS, "CourseContent")
|
| 471 |
CASE_KB_TEXT, CASE_PLACEHOLDER_ONLY = _kb_extract(CASE_PATHS, "CaseStudyContent")
|
| 472 |
|
| 473 |
-
|
| 474 |
# =====================================================
|
| 475 |
# DEMO ASSUMPTIONS
|
| 476 |
# =====================================================
|
|
@@ -541,7 +532,6 @@ def kb_block_for_mode(mode: str) -> str:
|
|
| 541 |
# fallback
|
| 542 |
return "[Knowledge Base]\n(none)"
|
| 543 |
|
| 544 |
-
|
| 545 |
# =====================================================
|
| 546 |
# LLM COMPLETION
|
| 547 |
# =====================================================
|
|
@@ -562,7 +552,6 @@ def complete_llm(messages, mode: str) -> str:
|
|
| 562 |
return r["choices"][0]["message"]["content"]
|
| 563 |
return r.choices[0].message.content
|
| 564 |
|
| 565 |
-
|
| 566 |
# =====================================================
|
| 567 |
# PER-MODE CHAT HISTORY SUPPORT
|
| 568 |
# =====================================================
|
|
@@ -570,7 +559,6 @@ def complete_llm(messages, mode: str) -> str:
|
|
| 570 |
def load_history_for_mode(selected_mode, histories):
|
| 571 |
return histories.get(selected_mode, [])
|
| 572 |
|
| 573 |
-
|
| 574 |
# =====================================================
|
| 575 |
# MAIN CHAT HANDLER
|
| 576 |
# =====================================================
|
|
@@ -645,7 +633,6 @@ def chat_user(message, visible_history, mode, upload_paths, histories, memory_st
|
|
| 645 |
# Clear textbox AND upload after sending
|
| 646 |
return new_history, "", None, histories, memory_state, flags_state
|
| 647 |
|
| 648 |
-
|
| 649 |
# =====================================================
|
| 650 |
# EXPORT & RESET
|
| 651 |
# =====================================================
|
|
@@ -665,7 +652,6 @@ def reset_chat(histories, memory_state, flags_state):
|
|
| 665 |
histories[k] = []
|
| 666 |
return [], None, "", histories, memory_state, flags_state
|
| 667 |
|
| 668 |
-
|
| 669 |
# =====================================================
|
| 670 |
# UI
|
| 671 |
# =====================================================
|
|
|
|
| 23 |
except Exception:
|
| 24 |
requests = None
|
| 25 |
|
|
|
|
| 26 |
# =====================================================
|
| 27 |
# CORE CONFIG
|
| 28 |
# =====================================================
|
|
|
|
| 38 |
os.makedirs(LOG_DIR, exist_ok=True)
|
| 39 |
LOG_PATH = os.path.join(LOG_DIR, "conversations.jsonl")
|
| 40 |
|
|
|
|
| 41 |
# =====================================================
|
| 42 |
# KNOWLEDGE BASE (CourseContent / CaseStudyContent)
|
| 43 |
# =====================================================
|
|
|
|
| 54 |
def _is_placeholder_name(name: str) -> bool:
|
| 55 |
return "placeholder" in (name or "").lower()
|
| 56 |
|
|
|
|
| 57 |
# =====================================================
|
| 58 |
# MODES (single customization surface) ****** CUSTOMIZE ******
|
| 59 |
# =====================================================
|
|
|
|
| 182 |
STUDENT_MODES = list(MODE_REGISTRY.keys())
|
| 183 |
DEFAULT_MODE = STUDENT_MODES[0] if STUDENT_MODES else "Course Detailer"
|
| 184 |
|
|
|
|
| 185 |
# =====================================================
|
| 186 |
# OPTIONAL WEB SEARCH (OFF by default) ****** CUSTOMIZE ******
|
| 187 |
# =====================================================
|
|
|
|
| 195 |
return ""
|
| 196 |
return ""
|
| 197 |
|
|
|
|
| 198 |
# =====================================================
|
| 199 |
# MEMORY + LOGGING
|
| 200 |
# =====================================================
|
|
|
|
| 217 |
with open(LOG_PATH, "a", encoding="utf-8") as f:
|
| 218 |
f.write(json.dumps(rec, ensure_ascii=False) + "\n")
|
| 219 |
|
|
|
|
| 220 |
# =====================================================
|
| 221 |
# OUTPUT CLEANING
|
| 222 |
# =====================================================
|
|
|
|
| 229 |
cleaned = _THINK_RE.sub("", text).strip()
|
| 230 |
return cleaned if cleaned else "(No visible answer returned. Please re-try.)"
|
| 231 |
|
|
|
|
| 232 |
# =====================================================
|
| 233 |
# FILE EXTRACTION
|
| 234 |
# =====================================================
|
|
|
|
| 405 |
badge = f"Attached: {', '.join(names[:MAX_FILES])}" if names else ""
|
| 406 |
return combined, meta_list, badge
|
| 407 |
|
|
|
|
| 408 |
# =====================================================
|
| 409 |
# STATIC KB INGESTION
|
| 410 |
# =====================================================
|
|
|
|
| 462 |
COURSE_KB_TEXT, COURSE_PLACEHOLDER_ONLY = _kb_extract(COURSE_PATHS, "CourseContent")
|
| 463 |
CASE_KB_TEXT, CASE_PLACEHOLDER_ONLY = _kb_extract(CASE_PATHS, "CaseStudyContent")
|
| 464 |
|
|
|
|
| 465 |
# =====================================================
|
| 466 |
# DEMO ASSUMPTIONS
|
| 467 |
# =====================================================
|
|
|
|
| 532 |
# fallback
|
| 533 |
return "[Knowledge Base]\n(none)"
|
| 534 |
|
|
|
|
| 535 |
# =====================================================
|
| 536 |
# LLM COMPLETION
|
| 537 |
# =====================================================
|
|
|
|
| 552 |
return r["choices"][0]["message"]["content"]
|
| 553 |
return r.choices[0].message.content
|
| 554 |
|
|
|
|
| 555 |
# =====================================================
|
| 556 |
# PER-MODE CHAT HISTORY SUPPORT
|
| 557 |
# =====================================================
|
|
|
|
| 559 |
def load_history_for_mode(selected_mode, histories):
|
| 560 |
return histories.get(selected_mode, [])
|
| 561 |
|
|
|
|
| 562 |
# =====================================================
|
| 563 |
# MAIN CHAT HANDLER
|
| 564 |
# =====================================================
|
|
|
|
| 633 |
# Clear textbox AND upload after sending
|
| 634 |
return new_history, "", None, histories, memory_state, flags_state
|
| 635 |
|
|
|
|
| 636 |
# =====================================================
|
| 637 |
# EXPORT & RESET
|
| 638 |
# =====================================================
|
|
|
|
| 652 |
histories[k] = []
|
| 653 |
return [], None, "", histories, memory_state, flags_state
|
| 654 |
|
|
|
|
| 655 |
# =====================================================
|
| 656 |
# UI
|
| 657 |
# =====================================================
|