Update app.py

app.py

@@ -1,57 +1,37 @@
 import gradio as gr
 from transformers import pipeline
-import time
 
-# मॉडल लोड करें (ChatGPT जैसा अनुभव के लिए आप facebook/blenderbot-400M-distill, gpt2, या google/gemma-2b-it भी ले सकते हैं)
-model_name = "facebook/blenderbot-400M-distill"
-chatbot = pipeline("conversational", model=model_name)
+# Hugging Face का कोई भी चैट मॉडल लोड करें (यहाँ facebook/blenderbot-400M-distill उदाहरण है)
+model = pipeline("conversational", model="facebook/blenderbot-400M-distill")
 
-# सिस्टम मैसेज/वेलकम मैसेज
-SYSTEM_PROMPT = """आप एक मददगार, विनम्र और ज्ञानवर्धक AI सहायक हैं। 
-आप छात्रों के सवालों का जवाब देते हैं और उनकी मदद करते हैं।"""
-
-# कस्टम थीम और लेआउट
-theme = gr.themes.Soft(
-    primary_hue="teal",
-    secondary_hue="teal",
-    neutral_hue="slate",
-    font=["Roboto", "ui-sans-serif", "sans-serif"],
-    font_mono=["Roboto Mono", "ui-monospace", "monospace"]
-)
-
-# चैट फंक्शन
-def chat_with_ai(message, chat_history):
-    # सिस्टम प्रॉम्प्ट को चैट हिस्ट्री में जोड़ें (यदि पहली बार चैट कर रहे हैं)
-    if not chat_history:
-        chat_history.append((None, "नमस्ते! मैं आपकी कैसे मदद कर सकता हूँ?"))
+# यह फंक्शन AI को प्रश्न भेजता है और उत्तर लेता है
+def qna_ai(question, chat_history):
+    # चैट हिस्ट्री को सिर्फ प्रश्न-उत्तर के फॉर्मेट में रखें
+    formatted_history = "\n".join([f"User: {q}\nAI: {a}" for q, a in chat_history])
+    prompt = f"{formatted_history}\nUser: {question}\nAI:"
     
-    # AI को मैसेज भेजें
-    conversation = chatbot(message)
-    ai_response = conversation.generated_responses[-1]
+    # AI से जवाब लें
+    response = model(prompt)
+    ai_answer = response.generated_responses[-1]
     
-    # चैट हिस्ट्री अपडेट करें
-    chat_history.append((message, ai_response))
+    # चैट हिस्ट्री को अपडेट करें
+    chat_history.append((question, ai_answer))
     return "", chat_history
 
-# साफ़ करने का फंक्शन
+# चैट हिस्ट्री साफ़ करने का फंक्शन
 def clear_chat():
     return [], []
 
-# एडवांस्ड UI बनाएं
-with gr.Blocks(theme=theme, title="AI चैटबोर्ड | छात्र सहायक") as demo:
-    gr.Markdown("# 🚀 AI चैटबोर्ड – छात्रों के लिए सहायक")
-    gr.Markdown("यहां आप अपने सवाल पूछ सकते हैं और AI आपको जवाब देगा!")
-    
-    chatbot = gr.Chatbot(label="चैट", height=500)
-    msg = gr.Textbox(label="आपका सवाल", placeholder="यहां लिखें...")
-    clear = gr.Button("साफ़ करें", variant="secondary")
-    
-    msg.submit(chat_with_ai, [msg, chatbot], [msg, chatbot])
-    clear.click(clear_chat, None, chatbot)
-    
-    # फुटर/क्रेडिट
-    gr.Markdown("---")
-    gr.Markdown("**बनाया गया: Gradio + Hugging Face + Transformers**")
+# एडवांस्ड चैट इंटरफेस बनाएं
+with gr.Blocks(title="छात्र सहायक AI", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🎓 छात्रों के लिए प्रश्न-उत्तर AI")
+    gr.Markdown("यहां अपना प्रश्न लिखें और AI आपको उत्तर देगा!")
 
-demo.launch(share=False)
+    chatbot = gr.Chatbot(label="चैट", height=400)
+    msg = gr.Textbox(label="प्रश्न", placeholder="यहां लिखें...")
+    clear = gr.Button("साफ़ करें")
+
+    msg.submit(qna_ai, [msg, chatbot], [msg, chatbot])
+    clear.click(clear_chat, None, chatbot)
 
+demo.launch()