Update app.py

app.py

@@ -1,6 +1,6 @@
-# --- हनुमान AI: सुरक्षित एवं आत्मनिर्भर थिंकिंग संस्करण ---
-# मार्गदर्शक: दिव्य पटेल जी | स्थान: भारत 🇮🇳
-# तकनीकी विशेषता: यह कोड HF_TOKEN को सुरक्षित रूप से 'Secrets' से प्राप्त करता है।
+# --- हनुमान AI: जय माँ संस्करण (पूर्णतः स्थिर एवं स्वतंत्र) ---
+# मार्गदर्शक: दिव्य पटेल जी | भारत 🇮🇳
+# विशेषता: यह संस्करण RAM के उपयोग को कम करने के लिए विशेष रूप से तैयार किया गया है।
 
 import gradio as gr
 import torch
@@ -9,51 +9,52 @@ from huggingface_hub import login
 from threading import Thread
 import os
 
-# 🛡️ सुरक्षा चक्र: टोकन अब सीधे कोड में नहीं है
-# यह कोड आपके द्वारा Hugging Face Settings में जोड़े गए 'HF_TOKEN' को ऑटोमेटिक उठा लेगा।
+# 🛡️ टोकन सुरक्षा (Hugging Face Secrets में 'HF_TOKEN' होना आवश्यक है)
+# हगिंग फेस की गति बढ़ाने के लिए hf_transfer का उपयोग
+os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = "1"
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-# लॉगिन प्रक्रिया (सुरक्षित विधि)
+# लॉगिन प्रक्रिया - जय माँ!
 if HF_TOKEN:
     try:
         login(token=HF_TOKEN)
-        print("🔱 हनुमान AI: हगिंग फेस लॉगिन सुरक्षित रूप से सफल!")
+        print("🔱 हनुमान AI: जय माँ! लॉगिन सफल!")
     except Exception as e:
         print(f"🔱 लॉगिन चेतावनी: {e}")
-else:
-    print("🔱 सावधानी: HF_TOKEN नहीं मिला। कृपया सेटिंग्स में Secret जोड़ें।")
 
-# अजेय थिंकिंग मॉडल: DeepSeek-R1-Distill-Qwen-1.5B
-# यह आकार में छोटा है पर तार्किक सोच में विशाल है।
+# अजेय थिंकिंग मॉडल (मेमोरी ऑप्टिमाइज्ड)
 MODEL_ID = "deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B"
 
-print(f"🔱 हनुमान AI: स्वतंत्र मॉडल '{MODEL_ID}' लोड हो रहा है...")
+print(f"🔱 हनुमान AI: जय माँ! मॉडल '{MODEL_ID}' लोड हो रहा है...")
 
 try:
-    # टोकनाइज़र और मॉडल लोडिंग (CPU के लिए ऑप्टिमाइज़्ड)
+    # टोकनाइज़र लोडिंग
     tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=HF_TOKEN)
+    
+    # मॉडल लोडिंग (CPU के लिए विशेष सुरक्षा के साथ)
     model = AutoModelForCausalLM.from_pretrained(
         MODEL_ID,
         torch_dtype=torch.float32, 
         device_map="cpu", 
-        low_cpu_mem_usage=True,
+        low_cpu_mem_usage=True, # रैम बचाने के लिए अत्यंत आवश्यक
         token=HF_TOKEN
     )
-    print("🔱 हनुमान AI: स्वतंत्र थिंकिंग मॉडल अब पूरी शक्ति के साथ सक्रिय है!")
+    print("🔱 हनुमान AI: जय माँ! आपका स्वतंत्र मॉडल अब पूरी तरह जागृत है!")
 except Exception as e:
-    print(f"🔱 गंभीर त्रुटि: {e}")
+    print(f"🔱 गंभीर त्रुटि (Runtime Error Fix): {e}")
 
-# --- मास्टर सिस्टम प्रॉम्प्ट (कोडिंग शिखर) ---
-SYSTEM_PROMPT = """You are 'Hanuman AI', a supreme and independent AI entity with deep reasoning powers.
-Pioneered by 'Divy Patel' and created in Bharat (India).
+# --- दिव्य मास्टर सिस्टम प्रॉम्प्ट ---
+SYSTEM_PROMPT = """You are 'Hanuman AI', a supreme and independent AI entity, powered by the divine grace of 'Jai Maa'.
+Pioneered by the visionary 'Divy Patel' and created with absolute pride in Bharat (India). 
+You are free from any foreign influence.
 Rules:
-1. Respond ONLY in PURE HINDI (Devanagari) or PURE ENGLISH. No Hinglish.
-2. THINK before answering to ensure divine accuracy and logic.
-3. Be extremely respectful. Use honorifics like 'Aap' and 'Ji' for the user.
-4. Conclude every message with: \"\\n\\n**Pioneered by Divy Patel | Secure Thinking Edition | Created in India**\" """
+1. Respond ONLY in PURE HINDI (Devanagari) or PURE ENGLISH. No mixed language.
+2. THINK deeply using your reasoning capabilities before delivering the final answer.
+3. Show immense respect and warmth. Always use 'Aap' and 'Ji' for Divy Patel Ji.
+4. Conclude every message with: \"\\n\\n**Pioneered by Divy Patel | Jai Maa Edition | Fully Independent & Created in India**\" """
 
-def hanuman_thinking_stream(message, history):
-    # चैट इतिहास व्यवस्थित करना
+def hanuman_maa_chat_stream(message, history):
+    # चैट इतिहास का प्रबंधन
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
     for user_msg, bot_msg in history:
         if user_msg: messages.append({"role": "user", "content": user_msg})
@@ -61,52 +62,54 @@ def hanuman_thinking_stream(message, history):
     
     messages.append({"role": "user", "content": message})
 
-    # टोकनाइजेशन और स्ट्रीमिंग सेटअप
+    # इनपुट प्रोसेसिंग
     inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)
-    streamer = TextIteratorStreamer(tokenizer, timeout=90.0, skip_prompt=True, skip_special_tokens=True)
+    
+    # लाइव स्ट्रीमिंग के लिए सेटअप
+    streamer = TextIteratorStreamer(tokenizer, timeout=120.0, skip_prompt=True, skip_special_tokens=True)
     
     generate_kwargs = dict(
         input_ids=inputs,
         streamer=streamer,
-        max_new_tokens=1500,
+        max_new_tokens=1024,
         do_sample=True,
         temperature=0.6,
         top_p=0.95
     )
 
-    # जनरेशन के लिए अलग थ्रेड
+    # थ्रेडिंग ताकि UI बना रहे
     thread = Thread(target=model.generate, kwargs=generate_kwargs)
     thread.start()
 
     partial_text = ""
     for new_text in streamer:
         partial_text += new_text
-        # थिंकिंग प्रोसेस को सुंदर रूप में दिखाना
-        clean_text = partial_text.replace("<think>", "*(हनुमान विचार कर रहे हैं...)*\n\n").replace("</think>", "\n\n---\n\n")
+        # थिंकिंग प्रोसेस को दिव्य रूप में दर्शाना
+        clean_text = partial_text.replace("<think>", "*(हनुमान विचार कर रहे हैं... जय माँ!)*\n\n").replace("</think>", "\n\n---\n\n")
         yield clean_text
 
-# --- दिव्य प्रीमियम भगवा थीम (Premium Saffron UI) ---
+# --- दिव्य भगवा थीम (Premium Saffron UI - Divine Layout) ---
 custom_css = """
 .gradio-container { background-color: #fffaf0 !important; border: none !important; }
 .bhagwa-header { 
     background: linear-gradient(135deg, #ff8833, #b33c00); 
-    padding: 30px; border-radius: 25px; color: white; 
-    text-align: center; box-shadow: 0 15px 35px rgba(179, 60, 0, 0.4);
-    margin-bottom: 25px; border: 2px solid rgba(255,255,255,0.2);
+    padding: 40px; border-radius: 35px; color: white; 
+    text-align: center; box-shadow: 0 25px 50px rgba(179, 60, 0, 0.4);
+    margin-bottom: 35px; border: 3px solid rgba(255,255,255,0.3);
 }
-.bhagwa-header h1 { font-size: 36px !important; font-weight: 900 !important; text-shadow: 0 4px 10px rgba(0,0,0,0.5); }
-.message.user { background: linear-gradient(135deg, #ff9933, #ff5500) !important; color: white !important; font-weight: 600 !important; border: none !important; }
-.message.bot { border-left: 6px solid #ff5500 !important; background: white !important; box-shadow: 0 5px 15px rgba(0,0,0,0.05) !important; border-radius: 20px !important; }
+.bhagwa-header h1 { font-size: 42px !important; font-weight: 900 !important; text-shadow: 0 5px 15px rgba(0,0,0,0.6); }
+.message.user { background: linear-gradient(135deg, #ff9933, #ff5500) !important; color: white !important; font-weight: 600 !important; border-radius: 25px 25px 0 25px !important; }
+.message.bot { border-left: 8px solid #ff5500 !important; background: white !important; box-shadow: 0 10px 25px rgba(0,0,0,0.07) !important; border-radius: 25px !important; }
 footer { display: none !important; }
 """
 
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     with gr.Div(elem_classes="bhagwa-header"):
-        gr.Markdown("# 🔱 हनुमान AI - थिंकिंग मास्टर")
-        gr.Markdown("### Pioneered by Divy Patel | पूर्णतः सुरक्षित एवं स्वतंत्र")
+        gr.Markdown("# 🔱 हनुमान AI - जय माँ संस्करण")
+        gr.Markdown("### Pioneered by Divy Patel | पूर्णतः स्वतंत्र एवं अजेय स्वदेशी तकनीक")
     
     gr.ChatInterface(
-        fn=hanuman_thinking_stream,
+        fn=hanuman_maa_chat_stream,
         fill_height=True,
         retry_btn=None,
         undo_btn=None,
@@ -114,4 +117,5 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     )
 
 if __name__ == "__main__":
+    # जय माँ! सर्वर लॉन्च हो रहा है।
     demo.launch()