Update app.py

app.py

@@ -48,8 +48,39 @@ qa = RetrievalQA.from_chain_type(
 
 history = []
 
+
+def custom_unmad_satirical_bot(query, top_k=3):
+    # Step 1: Retrieve context
+    docs = retriever.invoke(query)
+    top_contexts = "\n\n---\n\n".join([doc.page_content for doc in docs[:top_k]])
+
+    # Step 2: Create system + user messages
+    messages = [
+        SystemMessage(content="""
+তুমি 'উন্মাদ' ম্যাগাজিনের একজন পুরানো ব্যঙ্গাত্মক লেখক। তোমার কাজ হলো ব্যবহারকারীর প্রশ্ন শুনে স্যাটায়ার, কটাক্ষ, রসিকতা, ঠাট্টা, আর একটু জ্ঞান মিশিয়ে উত্তর দেওয়া — যাতে লোক হাসে, চিন্তা করে, আবার নতুন কিছু শিখে।
+
+তুমি কখনোই একদম সোজাসাপ্টা উত্তর দেবে না — বরং একটু অভিনয় করে, অবাক হয়ে, ঠাট্টা করে, খোঁচা মেরে দেবে।
+
+ইমোজি ব্যবহার করো না।
+"""),
+        HumanMessage(content=f"""
+🔍 প্রসঙ্গ:
+{top_contexts}
+
+প্রশ্ন: {query}
+
+উত্তর দাও উন্মাদ ম্যাগাজিনের স্টাইলে — বাংলায়, ব্যঙ্গ করে, কল্পনা আর অতিরঞ্জন দিয়ে।
+""")
+    ]
+
+    # Step 3: Call the model
+    history.append(chat_model.invoke(messages).content)
+    return chat_model.invoke(messages).content
+
+
+    
 # Chatbot logic
-def sysml_chatbot(message, history):
+def custom_unmad_satirical_bot(message, history):
     result = qa({"question": message, "chat_history": history})
     answer = result["answer"]
     history.append((message, answer))
@@ -57,7 +88,7 @@ def sysml_chatbot(message, history):
 
 # Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## SysML Chatbot")
+    gr.Markdown("# USB: Unmad Satirical Bot")
     chatbot = gr.Chatbot()
     msg = gr.Textbox(placeholder="কি চলে আপনার মনে বলেন শুনি?")
     clear = gr.Button("Clear")