Update app.py

app.py

@@ -22,8 +22,7 @@ def log(msg: str):
         with open(LOG_FILE, "a") as f:
             f.write(line + "\n")
     except Exception:
-        # If logging to file fails, we still want the process to continue
-        pass
+        pass  # tolerate logging failures
 
 # ---------------------------------------------------------------------------
 # 1.  Configuration constants
@@ -122,25 +121,22 @@ def build_prompt(raw_history: list[dict]) -> str:
 # ---------------------------------------------------------------------------
 def chat_fn(user_msg: str, display_history: list, state: dict):
     user_msg = strip(user_msg or "")
-    # Yield nothing if empty input
     if not user_msg:
         yield display_history, state
         return
 
-    # Input length check
     if len(user_msg) > MAX_INPUT_CH:
         display_history.append((user_msg, f"Input >{MAX_INPUT_CH} chars."))
         yield display_history, state
         return
 
-    # Model load error
     if MODEL_ERR:
         display_history.append((user_msg, MODEL_ERR))
         yield display_history, state
         return
 
     try:
-        # Add user to history and display
+        # record user
         state["raw"].append({"role": "user", "content": user_msg})
         display_history.append((user_msg, ""))
 
@@ -148,28 +144,25 @@ def chat_fn(user_msg: str, display_history: list, state: dict):
         start = time.time()
         partial = ""
 
-        # Stream tokens as they arrive
+        # stream chunks
         for chunk in generator(prompt):
             try:
                 new_text = strip(chunk.get("generated_text", ""))
-                # Truncate any hallucinated next-turn
                 if "User:" in new_text:
                     new_text = new_text.split("User:", 1)[0].strip()
                 partial += new_text
                 display_history[-1] = (user_msg, partial)
                 yield display_history, state
             except Exception:
-                # Skip malformed chunk but keep streaming
                 log("Malformed chunk:\n" + traceback.format_exc())
                 continue
 
-        # Finalize
+        # finalize
         full_reply = display_history[-1][1]
         state["raw"].append({"role": "assistant", "content": full_reply})
-        log(f"Reply in {time.time()-start:.2f}s ({len(full_reply)} chars)")
+        log(f"Reply in {time.time() - start:.2f}s ({len(full_reply)} chars)")
 
     except Exception:
-        # Catch-all for any unexpected errors in chat flow
         log("Unexpected chat_fn error:\n" + traceback.format_exc())
         err = "Apologies—an internal error occurred. Please try again."
         display_history[-1] = (user_msg, err)
@@ -183,9 +176,10 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
     gr.Markdown("### SchoolSpirit AI Chat")
 
     chatbot = gr.Chatbot(
-        value=[("", WELCOME_MSG)],
+        value=[{"role":"assistant","content":WELCOME_MSG}],
         height=480,
         label="SchoolSpirit AI",
+        type="messages",  # use the new messages format
     )
 
     state = gr.State(
@@ -199,12 +193,12 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue")) as demo:
         txt = gr.Textbox(placeholder="Type your question here…", show_label=False, scale=4, lines=1)
         send_btn = gr.Button("Send", variant="primary")
 
-    # Enable streaming updates in the UI
-    send_btn.click(chat_fn, inputs=[txt, chatbot, state], outputs=[chatbot, state], stream=True)
-    txt.submit(chat_fn, inputs=[txt, chatbot, state], outputs=[chatbot, state], stream=True)
+    # Use streaming=True (not stream) per Gradio API
+    send_btn.click(chat_fn, inputs=[txt, chatbot, state], outputs=[chatbot, state], streaming=True)
+    txt.submit(chat_fn, inputs=[txt, chatbot, state], outputs=[chatbot, state], streaming=True)
 
 if __name__ == "__main__":
     try:
         demo.launch()
-    except Exception as exc:
-        log(f"UI launch error:\n{traceback.format_exc()}")
+    except Exception:
+        log("UI launch error:\n" + traceback.format_exc())