Upload app.py with huggingface_hub

app.py

@@ -57,7 +57,8 @@ def chat(message, history, model):
     history = history or []
     prefix = {"Agni": "🔥 ", "Vayu": "🌊 ", "Akash": "🌌 "}.get(model, "🔥 ")
     response = get_response(message)
-    history.append((message, prefix + response))
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": prefix + response})
     return history, ""
 
 with gr.Blocks(title="Devrayog AI") as demo:
@@ -67,19 +68,24 @@ with gr.Blocks(title="Devrayog AI") as demo:
     with gr.Row():
         with gr.Column(scale=3):
             model = gr.Radio(["Agni", "Vayu", "Akash"], value="Agni", label="Model")
-            chatbot = gr.Chatbot(height=350, show_label=False)
+            chatbot = gr.Chatbot(height=350, show_label=False, type="messages")
             with gr.Row():
                 msg = gr.Textbox(placeholder="Ask about SSB, NDA, TAT...", scale=4, container=False, autofocus=True)
                 send = gr.Button("Send", scale=1, variant="primary")
             with gr.Row():
-                gr.Button("🎯 SSB", size="sm").click(lambda h: chat("What is SSB process?", h, "Agni"), inputs=[chatbot], outputs=[chatbot, msg])
-                gr.Button("📚 NDA", size="sm").click(lambda h: chat("What is NDA?", h, "Agni"), inputs=[chatbot], outputs=[chatbot, msg])
-                gr.Button("💡 Tips", size="sm").click(lambda h: chat("SSB tips", h, "Agni"), inputs=[chatbot], outputs=[chatbot, msg])
-                gr.Button("⭐ OLQs", size="sm").click(lambda h: chat("15 OLQs", h, "Agni"), inputs=[chatbot], outputs=[chatbot, msg])
+                b1 = gr.Button("🎯 SSB", size="sm")
+                b2 = gr.Button("📚 NDA", size="sm")
+                b3 = gr.Button("💡 Tips", size="sm")
+                b4 = gr.Button("⭐ OLQs", size="sm")
         with gr.Column(scale=1):
-            gr.Markdown("### Models\n🔥 **Agni** - Fast\n🌊 **Vayu** - Creative\n🌌 **Agni** - Powerful\n\n### Topics\n• SSB Process\n• TAT/WAT/SRT\n• NDA Exam\n• 15 OLQs\n• GTO Tasks")
+            gr.Markdown("### Models\n🔥 **Agni** - Fast\n🌊 **Vayu** - Creative\n🌌 **Akash** - Powerful\n\n### Topics\n• SSB Process\n• TAT/WAT/SRT\n• NDA Exam\n• 15 OLQs\n• GTO Tasks")
     
     send.click(fn=chat, inputs=[msg, chatbot, model], outputs=[chatbot, msg])
     msg.submit(fn=chat, inputs=[msg, chatbot, model], outputs=[chatbot, msg])
+    
+    b1.click(fn=lambda h, m: chat("What is SSB process?", h, m), inputs=[chatbot, model], outputs=[chatbot, msg])
+    b2.click(fn=lambda h, m: chat("What is NDA?", h, m), inputs=[chatbot, model], outputs=[chatbot, msg])
+    b3.click(fn=lambda h, m: chat("SSB tips", h, m), inputs=[chatbot, model], outputs=[chatbot, msg])
+    b4.click(fn=lambda h, m: chat("15 OLQs", h, m), inputs=[chatbot, model], outputs=[chatbot, msg])
 
 demo.launch(server_name="0.0.0.0", server_port=7860)