Update app.py

app.py

@@ -9,7 +9,6 @@ from typing import List, Tuple, Dict, Optional
 
 import gradio as gr
 
-# Optional readers
 try:
     from pypdf import PdfReader
 except Exception:
@@ -21,9 +20,6 @@ except Exception:
     docx = None
 
 
-# =========================================================
-# CONFIG
-# =========================================================
 APP_TITLE = "BrainChat"
 APP_SUBTITLE = "Neurology & neuroanatomy tutor"
 NOT_FOUND_TEXT = "Not found in the course material."
@@ -48,9 +44,6 @@ STOPWORDS = {
 }
 
 
-# =========================================================
-# HELPERS
-# =========================================================
 def normalize_spaces(text: str) -> str:
     return re.sub(r"\s+", " ", text).strip()
 
@@ -64,8 +57,7 @@ def split_sentences(text: str) -> List[str]:
     text = normalize_spaces(text)
     if not text:
         return []
-    parts = re.split(r"(?<=[.!?])\s+", text)
-    return [p.strip() for p in parts if p.strip()]
+    return [p.strip() for p in re.split(r"(?<=[.!?])\s+", text) if p.strip()]
 
 
 def chunk_text(text: str, chunk_size: int = 900, overlap: int = 150) -> List[str]:
@@ -170,10 +162,7 @@ def file_to_data_uri(path: Optional[Path]) -> Optional[str]:
         data = path.read_bytes()
         mime, _ = mimetypes.guess_type(str(path))
         if mime is None:
-            if path.suffix.lower() == ".svg":
-                mime = "image/svg+xml"
-            else:
-                mime = "image/png"
+            mime = "image/svg+xml" if path.suffix.lower() == ".svg" else "image/png"
         encoded = base64.b64encode(data).decode("utf-8")
         return f"data:{mime};base64,{encoded}"
     except Exception:
@@ -197,9 +186,6 @@ LOGO_FILE = find_asset([
 LOGO_URI = file_to_data_uri(LOGO_FILE)
 
 
-# =========================================================
-# KNOWLEDGE BASE
-# =========================================================
 class LocalKnowledgeBase:
     def __init__(self) -> None:
         self.docs: List[Dict] = []
@@ -218,13 +204,12 @@ class LocalKnowledgeBase:
         self.docs.append({"doc_id": doc_id, "source": source_name, "text": text})
 
         for idx, chunk in enumerate(chunk_text(text)):
-            tokens = set(tokenize(chunk))
             self.chunks.append({
                 "doc_id": doc_id,
                 "source": source_name,
                 "chunk_id": idx,
                 "text": chunk,
-                "tokens": tokens
+                "tokens": set(tokenize(chunk))
             })
 
     def load_from_directories(self) -> None:
@@ -251,7 +236,7 @@ class LocalKnowledgeBase:
                 if text.strip():
                     self.add_document(path.name, text)
 
-    def search(self, query: str, top_k: int = 5) -> List[Dict]:
+    def search(self, query: str, top_k: int = 6) -> List[Dict]:
         q_tokens = set(tokenize(query))
         if not q_tokens:
             return []
@@ -261,7 +246,6 @@ class LocalKnowledgeBase:
             overlap = len(q_tokens.intersection(item["tokens"]))
             if overlap == 0:
                 continue
-
             score = overlap / max(1, len(q_tokens))
             scored.append((score, item))
 
@@ -289,10 +273,7 @@ KB = LocalKnowledgeBase()
 KB.load_from_directories()
 
 
-# =========================================================
-# CONTENT BUILDERS
-# =========================================================
-def collect_context(hits: List[Dict], max_chars: int = 2000) -> Tuple[str, List[str]]:
+def collect_context(hits: List[Dict], max_chars: int = 2400) -> Tuple[str, List[str]]:
     if not hits:
         return "", []
 
@@ -326,13 +307,7 @@ def simple_definition_style(text: str, short: bool = False) -> str:
     sentences = split_sentences(text)
     if not sentences:
         return NOT_FOUND_TEXT
-
-    if short:
-        chosen = sentences[:3]
-        return "\n\n".join(chosen)
-
-    chosen = sentences[:6]
-    return "\n\n".join(chosen)
+    return "\n\n".join(sentences[:3] if short else sentences[:6])
 
 
 def detailed_teaching_style(query: str, text: str) -> str:
@@ -340,11 +315,11 @@ def detailed_teaching_style(query: str, text: str) -> str:
     if not sentences:
         return NOT_FOUND_TEXT
 
+    out = [f"**Topic:** {query.strip()}"]
+
     first = sentences[:2]
     rest = sentences[2:6]
 
-    out = [f"**Topic:** {query.strip()}"]
-
     if first:
         out.append("\n**Simple explanation:**")
         out.append(" ".join(first))
@@ -355,7 +330,7 @@ def detailed_teaching_style(query: str, text: str) -> str:
             out.append(f"- {s}")
 
     out.append("\n**Why this matters:**")
-    out.append("This concept is important because it helps students connect theory with brain structure, function, and clinical understanding.")
+    out.append("This concept is important because it helps students connect structure, function, and clinical understanding.")
 
     return "\n".join(out)
 
@@ -379,7 +354,7 @@ def build_flashcards(query: str, text: str, count: int = 5) -> str:
 
     out = ["**Flash Cards**"]
     for i, term in enumerate(terms, 1):
-        meaning = sentences[min(i - 1, len(sentences) - 1)] if sentences else f"{term} is an important concept from the material."
+        meaning = sentences[min(i - 1, len(sentences) - 1)] if sentences else f"{term} is an important concept."
         out.append(f"\n**Card {i}**")
         out.append(f"- **Term:** {term}")
         out.append(f"- **Meaning:** {meaning}")
@@ -394,19 +369,17 @@ def build_case_study(query: str, text: str) -> str:
 
     return (
         "**Case Study**\n\n"
-        f"A student is reviewing a patient-related topic connected to **{query}**. "
-        f"While studying, the student learns that: {base}\n\n"
+        f"A student is studying **{query}**. From the material, the student learns that: {base}\n\n"
         "**Scenario:**\n"
-        "A patient comes with symptoms that may involve this part of the nervous system. "
+        "A patient shows symptoms that may involve this part of the nervous system. "
         "The student must explain what structure or concept is involved, what it normally does, "
         "and what may happen when it is affected.\n\n"
         "**Questions to think about:**\n"
         "1. What is the main structure or idea here?\n"
         "2. What is its normal function?\n"
-        "3. What symptoms may appear if it is damaged or disturbed?\n"
+        "3. What symptoms may appear if it is damaged?\n"
         "4. Why is this topic important in neuroanatomy or neurology?\n\n"
-        "**Teacher-style note:**\n"
-        "This case study is for understanding and classroom learning, not for real medical diagnosis."
+        "**Note:** This is for classroom learning, not real diagnosis."
     )
 
 
@@ -429,7 +402,7 @@ def build_quiz(query: str, text: str, n_questions: int = 5) -> str:
     out = [f"**Quiz: {query.strip()}**"]
     for i, w in enumerate(unique_words[:n_questions], 1):
         out.append(f"{i}. What is **{w}**?")
-        out.append(f"{i}.a Why is **{w}** important?")
+        out.append(f"{i}a. Why is **{w}** important?")
     return "\n".join(out)
 
 
@@ -448,26 +421,19 @@ def build_revision_questions(query: str, text: str, n_questions: int = 5) -> str
             out.append(f"{i}. Give one example related to {query.strip()}.")
         else:
             if sentences:
-                out.append(f"{i}. Based on the material, explain this statement: “{sentences[min(i-5, len(sentences)-1)]}”")
+                out.append(f'{i}. Explain this statement: "{sentences[min(i-5, len(sentences)-1)]}"')
             else:
                 out.append(f"{i}. Write short notes on {query.strip()}.")
 
     return "\n".join(out)
 
 
-def make_learning_output(
-    query: str,
-    learning_mode: str,
-    tutor_mode: str,
-    quiz_questions: str,
-    hits: List[Dict]
-) -> Tuple[str, List[str]]:
+def make_learning_output(query: str, learning_mode: str, quiz_questions: str, hits: List[Dict]) -> Tuple[str, List[str]]:
     if not hits:
         return NOT_FOUND_TEXT, []
 
     context, sources = collect_context(hits, max_chars=2400)
     mode = (learning_mode or "Detailed Explanation").strip().lower()
-
     qn = 5 if str(quiz_questions).lower() == "auto" else int(quiz_questions)
 
     if mode == "normal answer":
@@ -504,97 +470,86 @@ def format_answer(answer_text: str, sources: List[str], show_sources: bool) -> s
     return answer_text
 
 
-def get_answer_and_sources(
-    message: str,
-    learning_mode: str,
-    tutor_mode: str,
-    answer_language: str,
-    quiz_questions: str
-) -> Tuple[str, List[str]]:
+def get_answer_and_sources(message: str, learning_mode: str, quiz_questions: str) -> Tuple[str, List[str]]:
     msg = (message or "").strip()
     if not msg:
         return "Please type a question.", []
 
     lower_msg = msg.lower().strip()
-
     if lower_msg in {"hi", "hello", "hey"}:
         return (
             "Hello. Ask me anything from your neurology or neuroanatomy material. "
-            "You can also choose a learning mode such as flash cards, case study, quiz, or detailed explanation.",
+            "You can choose flash cards, case study, quiz, short explanation, or detailed explanation.",
             []
         )
 
     hits = KB.search(msg, top_k=6)
-    return make_learning_output(msg, learning_mode, tutor_mode, quiz_questions, hits)
+    return make_learning_output(msg, learning_mode, quiz_questions, hits)
 
 
-# =========================================================
-# UI
-# =========================================================
 CUSTOM_CSS = """
 :root{
-    --bg1: #fffdf3;
-    --bg2: #fff4fb;
-    --panel: #ffffff;
-    --text: #2f2a4d;
-    --muted: #736d92;
-    --primary: #7c3aed;
-    --secondary: #ec4899;
-    --third: #f59e0b;
-    --mint: #6ee7b7;
-    --sky: #67e8f9;
-    --border: #eadcff;
+    --wa-bg: #efeae2;
+    --wa-panel: #f7f5f2;
+    --wa-header: #075e54;
+    --wa-header-2: #128c7e;
+    --wa-text: #1f2c34;
+    --wa-muted: #667781;
+    --wa-border: #d8e0dc;
+    --wa-user: #d9fdd3;
+    --wa-bot: #ffffff;
+    --wa-btn: #25d366;
+    --wa-btn-dark: #128c7e;
+    --wa-input: #ffffff;
 }
 
 html, body, .gradio-container {
-    background: linear-gradient(180deg, #fffdf3 0%, #fff7fb 55%, #fff2d7 100%) !important;
-    color: var(--text) !important;
+    background: linear-gradient(180deg, #e6ddd4 0%, #efeae2 100%) !important;
+    color: var(--wa-text) !important;
     font-family: "Segoe UI", Arial, sans-serif !important;
 }
 
 #main_shell {
-    max-width: 1180px;
+    max-width: 1100px;
     margin: 18px auto;
-    padding: 0 12px 18px 12px;
+    padding: 0 10px 18px 10px;
 }
 
 #topbar {
-    background: linear-gradient(90deg, #7c3aed 0%, #ec4899 55%, #f4d03f 100%);
-    border-radius: 30px;
-    padding: 18px 20px;
-    box-shadow: 0 12px 28px rgba(167, 85, 190, 0.22);
-    border: 2px solid rgba(255,255,255,0.7);
-    margin-bottom: 18px;
+    background: linear-gradient(90deg, #075e54 0%, #128c7e 100%);
+    border-radius: 22px;
+    padding: 16px 18px;
+    box-shadow: 0 10px 22px rgba(7, 94, 84, 0.20);
+    margin-bottom: 16px;
 }
 
 #brand_row {
     display: flex;
     align-items: center;
-    gap: 16px;
+    gap: 14px;
 }
 
 #brand_logo {
-    width: 78px;
-    height: 78px;
-    border-radius: 20px;
+    width: 68px;
+    height: 68px;
+    border-radius: 16px;
     object-fit: contain;
-    background: rgba(255,255,255,0.96);
-    padding: 6px;
-    box-shadow: 0 6px 18px rgba(0,0,0,0.12);
+    background: white;
+    padding: 5px;
+    box-shadow: 0 4px 10px rgba(0,0,0,0.12);
 }
 
 #brand_fallback {
-    width: 78px;
-    height: 78px;
-    border-radius: 20px;
+    width: 68px;
+    height: 68px;
+    border-radius: 16px;
     display: flex;
     align-items: center;
     justify-content: center;
-    background: rgba(255,255,255,0.96);
-    color: #7c3aed;
-    font-size: 28px;
+    background: white;
+    color: #075e54;
+    font-size: 26px;
     font-weight: 800;
-    box-shadow: 0 6px 18px rgba(0,0,0,0.12);
 }
 
 #brand_title {
@@ -607,41 +562,36 @@ html, body, .gradio-container {
 
 #brand_subtitle {
     font-size: 15px;
-    color: #fffdfd;
+    color: #dcf8c6;
     font-weight: 600;
     margin-top: 3px;
 }
 
 #settings_card, #chat_card {
-    background: rgba(255,255,255,0.92) !important;
-    border: 2px solid #eadcff !important;
-    border-radius: 24px !important;
-    box-shadow: 0 10px 24px rgba(120, 100, 180, 0.08);
-}
-
-#chatbot {
-    background: linear-gradient(180deg, #fffefb 0%, #fff7ff 100%) !important;
+    background: rgba(247,245,242,0.96) !important;
+    border: 1px solid var(--wa-border) !important;
     border-radius: 20px !important;
+    box-shadow: 0 8px 20px rgba(80, 90, 90, 0.08);
 }
 
-#chatbot .message.user {
-    background: linear-gradient(90deg, #8b5cf6, #ec4899) !important;
-    color: white !important;
-    border-radius: 18px !important;
-}
-
-#chatbot .message.bot {
-    background: linear-gradient(180deg, #fff8d9 0%, #fffdf3 100%) !important;
-    color: #2f2a4d !important;
-    border: 1px solid #f3df8c !important;
+#chatbot {
+    background:
+        linear-gradient(rgba(239,234,226,0.94), rgba(239,234,226,0.94)),
+        repeating-linear-gradient(
+            45deg,
+            rgba(255,255,255,0.08) 0px,
+            rgba(255,255,255,0.08) 12px,
+            rgba(0,0,0,0.02) 12px,
+            rgba(0,0,0,0.02) 24px
+        ) !important;
     border-radius: 18px !important;
 }
 
 textarea, input, .wrap textarea {
     border-radius: 18px !important;
-    border: 2px solid #e3d8ff !important;
-    background: #ffffff !important;
-    color: var(--text) !important;
+    border: 1px solid #d5ddd9 !important;
+    background: var(--wa-input) !important;
+    color: var(--wa-text) !important;
 }
 
 button {
@@ -652,37 +602,27 @@ button {
 }
 
 #send_btn {
-    background: linear-gradient(90deg, #7c3aed, #ec4899) !important;
+    background: linear-gradient(90deg, #25d366, #128c7e) !important;
     color: white !important;
 }
 
 #clear_btn {
-    background: linear-gradient(90deg, #fde047, #facc15) !important;
-    color: #4b3b00 !important;
-}
-
-#upload_btn {
-    background: linear-gradient(90deg, #67e8f9, #6ee7b7) !important;
-    color: #17324d !important;
+    background: #ffffff !important;
+    color: #075e54 !important;
+    border: 1px solid #bfd6cf !important;
 }
 
 #reload_btn {
-    background: linear-gradient(90deg, #f9f5ff, #fff2fb) !important;
-    color: #473a69 !important;
-    border: 2px solid #e6d7ff !important;
+    background: #e7f7ee !important;
+    color: #075e54 !important;
+    border: 1px solid #bfe6d0 !important;
 }
 
 .small_hint {
-    color: var(--muted);
+    color: var(--wa-muted);
     font-size: 13px;
 }
 
-.gr-file, .gr-file * {
-    background: #fffaf0 !important;
-    color: #2f2a4d !important;
-    border-color: #f3df8c !important;
-}
-
 footer {
     display: none !important;
 }
@@ -708,19 +648,20 @@ def build_header_html() -> str:
     """
 
 
-def respond(message, history, learning_mode, tutor_mode, answer_language, quiz_questions, show_sources):
+def respond(message, history, learning_mode, quiz_questions, show_sources):
     history = history or []
 
     answer_text, sources = get_answer_and_sources(
         message=message,
         learning_mode=learning_mode,
-        tutor_mode=tutor_mode,
-        answer_language=answer_language,
         quiz_questions=quiz_questions
     )
 
     final_text = format_answer(answer_text, sources, show_sources)
-    history.append((message, final_text))
+
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": final_text})
+
     return history, ""
 
 
@@ -735,27 +676,7 @@ def reload_materials():
     return f"Materials reloaded. Loaded {len(KB.docs)} document(s)."
 
 
-def upload_files(files):
-    if not files:
-        return "No file uploaded."
-
-    added = 0
-    for f in files:
-        try:
-            path = Path(f.name)
-            text = extract_text_from_file(path)
-            if text.strip():
-                KB.add_document(path.name, text)
-                added += 1
-        except Exception:
-            continue
-
-    if added == 0:
-        return "No readable text found in uploaded files."
-    return f"{added} file(s) added successfully."
-
-
-with gr.Blocks(css=CUSTOM_CSS, title=APP_TITLE) as demo:
+with gr.Blocks(title=APP_TITLE) as demo:
     with gr.Column(elem_id="main_shell"):
         gr.HTML(build_header_html())
 
@@ -774,18 +695,6 @@ with gr.Blocks(css=CUSTOM_CSS, title=APP_TITLE) as demo:
                 label="Learning Mode"
             )
 
-            tutor_mode = gr.Dropdown(
-                ["Brief", "Detailed"],
-                value="Detailed",
-                label="Tutor Mode"
-            )
-
-            answer_language = gr.Dropdown(
-                ["Auto", "English", "Spanish"],
-                value="Auto",
-                label="Answer Language"
-            )
-
             quiz_questions = gr.Dropdown(
                 ["Auto", "5", "10"],
                 value="Auto",
@@ -804,26 +713,18 @@ with gr.Blocks(css=CUSTOM_CSS, title=APP_TITLE) as demo:
 
         with gr.Column(elem_id="chat_card"):
             chatbot = gr.Chatbot(
-                height=470,
+                height=500,
                 elem_id="chatbot",
-                show_label=False
+                show_label=False,
+                type="messages"
             )
 
-            with gr.Row():
-                file_input = gr.File(
-                    file_count="multiple",
-                    file_types=[".txt", ".md", ".pdf", ".docx"],
-                    label="Add material files (optional)",
-                    scale=2
-                )
-                msg = gr.Textbox(
-                    placeholder="Ask a question about neurology or neuroanatomy...",
-                    show_label=False,
-                    scale=6
-                )
+            msg = gr.Textbox(
+                placeholder="Type your question here...",
+                show_label=False
+            )
 
             with gr.Row():
-                upload_btn = gr.Button("Upload Files", elem_id="upload_btn")
                 reload_btn = gr.Button("Reload Materials", elem_id="reload_btn")
                 clear_btn = gr.Button("Clear Chat", elem_id="clear_btn")
                 send_btn = gr.Button("Send", elem_id="send_btn")
@@ -832,13 +733,13 @@ with gr.Blocks(css=CUSTOM_CSS, title=APP_TITLE) as demo:
 
     send_btn.click(
         respond,
-        inputs=[msg, chatbot, learning_mode, tutor_mode, answer_language, quiz_questions, show_sources],
+        inputs=[msg, chatbot, learning_mode, quiz_questions, show_sources],
         outputs=[chatbot, msg]
     )
 
     msg.submit(
         respond,
-        inputs=[msg, chatbot, learning_mode, tutor_mode, answer_language, quiz_questions, show_sources],
+        inputs=[msg, chatbot, learning_mode, quiz_questions, show_sources],
         outputs=[chatbot, msg]
     )
 
@@ -852,11 +753,5 @@ with gr.Blocks(css=CUSTOM_CSS, title=APP_TITLE) as demo:
         outputs=[status_box]
     )
 
-    upload_btn.click(
-        upload_files,
-        inputs=[file_input],
-        outputs=[status_box]
-    )
-
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(css=CUSTOM_CSS)