Update app.py

app.py

@@ -38,21 +38,12 @@ llm = ChatOpenAI(
     openai_api_key=OPENAI_API_KEY
 )
 
-# Build RetrievalQA chain for backup QA route (optional)
-qa = RetrievalQA.from_chain_type(
-    llm=llm,
-    chain_type="stuff",
-    retriever=vectorstore.as_retriever()
-)
-
-# Satirical Unmad-style QA function
+# Satirical QA function
 def custom_unmad_satirical_bot(message, history, top_k=3):
-    # Step 1: Retrieve RAG context
     retriever = vectorstore.as_retriever()
     docs = retriever.invoke(message)
     top_contexts = "\n\n---\n\n".join([doc.page_content for doc in docs[:top_k]])
 
-    # Step 2: Create prompt messages
     messages = [
         SystemMessage(content="""
 তুমি 'উন্মাদ' ম্যাগাজিনের একজন পুরানো ব্যঙ্গাত্মক লেখক। তোমার কাজ হলো ব্যবহারকারীর প্রশ্ন শুনে স্যাটায়ার, কটাক্ষ, রসিকতা, ঠাট্টা, আর একটু জ্ঞান মিশিয়ে উত্তর দেওয়া — যাতে লোক হাসে, চিন্তা করে, আবার নতুন কিছু শিখে।
@@ -62,28 +53,35 @@ def custom_unmad_satirical_bot(message, history, top_k=3):
         HumanMessage(content=f"""
 🔍 প্রসঙ্গ:
 {top_contexts}
-
 প্রশ্ন: {message}
-
 উত্তর দাও উন্মাদ ম্যাগাজিনের স্টাইলে — বাংলায়, ব্যঙ্গ করে, কল্পনা আর অতিরঞ্জন দিয়ে।
 """)
     ]
 
-    # Step 3: Generate response
     response = llm.invoke(messages).content
     history.append((message, response))
     return "", history
 
 # Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("# 🧠 USB: Unmad Satirical Bot (উন্মাদীয় বট)")
+with gr.Blocks(css=".gradio-container {padding-top: 50px;}") as demo:
+    gr.Markdown("# 🧠 USB: Unmad Satirical Bot (উন্মাদীয় বট)", elem_id="title")
+
+    # Insert Unmad image
+    with gr.Row():
+        gr.Image("images/unmad_pic.png", label="Unmad Mood", width=320, show_label=False)
+
     chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="কি চলে আপনার মনে বলেন শুনি?")
-    clear = gr.Button("Clear")
+    
+    with gr.Row():
+        msg = gr.Textbox(placeholder="কি চলে আপনার মনে বলেন শুনি?", scale=8, show_label=False)
+        send = gr.Button("Send", variant="primary", scale=1)
 
+    clear = gr.Button("Clear")
     state = gr.State([])
 
+    # Connect interaction
     msg.submit(custom_unmad_satirical_bot, [msg, state], [msg, chatbot])
+    send.click(custom_unmad_satirical_bot, [msg, state], [msg, chatbot])
     clear.click(lambda: ([], ""), None, [chatbot, msg])
 
 if __name__ == "__main__":