Update app.py


app.py

@@ -1,10 +1,113 @@
 import gradio as gr
 from transformers import pipeline
+import time
 
-# Load model
+# Load Bangla model
 bangla_ai = pipeline("text2text-generation", model="csebuetnlp/banglat5")
 
-def chat(message):
-    return bangla_ai(message, max_length=100)[0]['generated_text']
+# Bangladesh Flag Colors
+BD_GREEN = "#006a4e"
+BD_RED = "#f42a41"
 
-gr.ChatInterface(chat, title="NOTUNBANGLA AI").launch()
+# Custom CSS with Animations
+custom_css = f"""
+footer {{visibility: hidden}}
+@import url('https://fonts.maateen.me/kalpurush/font.css');
+.bangla-font * {{
+    font-family: "Kalpurush", "Siyam Rupali", sans-serif !important;
+}}
+
+/* Welcome Animation */
+@keyframes welcome {{
+    0% {{ transform: scale(0.8); opacity: 0; }}
+    100% {{ transform: scale(1); opacity: 1; }}
+}}
+.welcome-anim {{
+    animation: welcome 1s ease-out;
+}}
+
+/* Thinking Dots */
+.thinking-dots {{
+    display: inline-block;
+}}
+.thinking-dots span {{
+    display: inline-block;
+    width: 8px;
+    height: 8px;
+    border-radius: 50%;
+    background: {BD_GREEN};
+    margin: 0 2px;
+    opacity: 0;
+}}
+.thinking-dots span:nth-child(1) {{
+    animation: dot-pulse 1.5s infinite;
+}}
+.thinking-dots span:nth-child(2) {{
+    animation: dot-pulse 1.5s infinite 0.2s;
+}}
+.thinking-dots span:nth-child(3) {{
+    animation: dot-pulse 1.5s infinite 0.4s;
+}}
+@keyframes dot-pulse {{
+    0%, 100% {{ opacity: 0.3; transform: translateY(0); }}
+    50% {{ opacity: 1; transform: translateY(-5px); }}
+}}
+
+/* Button Style */
+button {{
+    background: {BD_GREEN} !important;
+    color: white !important;
+}}
+button:hover {{
+    background: {BD_RED} !important;
+}}
+"""
+
+def chat(message, history):
+    # Show thinking dots
+    yield "প্রস্তুত হচ্ছে... <div class='thinking-dots'><span></span><span></span><span></span></div>"
+    
+    # Simulate processing time
+    time.sleep(1.5)  # Remove this in production
+    
+    # Get AI response
+    response = bangla_ai(message, max_length=150)[0]['generated_text']
+    yield response
+
+with gr.Blocks(
+    title="নতুন বাংলা AI",
+    css=custom_css,
+    theme=gr.themes.Default(primary_hue="green")
+) as app:
+    
+    # Welcome Animation
+    with gr.Row():
+        with gr.Column():
+            gr.HTML("""
+            <div class="welcome-anim" style="text-align:center">
+                <img src="logo.png" width="120" style="margin-bottom:10px">
+                <h1 class="bangla-font" style="color:#006a4e; margin-bottom:0">নতুন বাংলা AI</h1>
+                <p class="bangla-font" style="color:#f42a41">জাতীয় জ্ঞান সহায়ক</p>
+            </div>
+            """)
+    
+    # Chat Interface
+    gr.ChatInterface(
+        fn=chat,
+        chatbot=gr.Chatbot(
+            avatar_images=(None, "logo.png"),
+            bubble_full_width=False,
+            render_markdown=True
+        ),
+        textbox=gr.Textbox(
+            placeholder="এখানে বাংলায় প্রশ্ন লিখুন...",
+            label=""
+        ),
+        submit_btn=gr.Button("প্রেরণ", variant="primary"),
+        examples=[
+            ["বাংলাদেশের স্বাধীনতা দিবস কবে?"],
+            ["রবীন্দ্রনাথ ঠাকুর কে ছিলেন?"]
+        ]
+    )
+
+app.launch()