Update app.py

app.py

@@ -1,6 +1,6 @@
-# --- हनुमान AI: जेम्मा मास्टर (अजेय और स्वतंत्र) ---
-# मार्गदर्शक: दिव्य पटेल जी | भारत 🇮🇳
-# कोडिंग शिखर: यह संस्करण मेमोरी और गेटेड एक्सेस के लिए पूर्णतः अनुकूलित है।
+# --- हनुमान AI: आत्मनिर्भर एवं फ्री स्पेस संस्करण ---
+# मार्गदर्शक: दिव्य पटेल जी | स्थान: भारत 🇮🇳
+# तकनीकी विशेषता: यह संस्करण हगिंग फेस के 16GB फ्री RAM के लिए पूर्णतः सुरक्षित है।
 
 import gradio as gr
 import torch
@@ -9,66 +9,72 @@ from huggingface_hub import login
 from threading import Thread
 import os
 
-# दिव्य पटेल जी का टोकन
-# सुझाव: इसे Hugging Face Space की Settings -> Secrets में 'HF_TOKEN' नाम से जोड़ना सबसे सुरक्षित है।
-HF_TOKEN = "Hf_ocaQWpMXbnuMhBjNdjuPQtVxZwPrRlXwjU"
+# 🛡️ टोकन सुरक्षा (टोकन को 'Secret' से उठाना सर्वोत्तम है)
+# दिव्य जी, कृपया हगिंग फेस Space की Settings -> Secrets में 'HF_TOKEN' जोड़ें।
+HF_TOKEN = os.getenv("HF_TOKEN") or "Hf_ocaQWpMXbnuMhBjNdjuPQtVxZwPrRlXwjU"
 
 # लॉगिन प्रक्रिया
 if HF_TOKEN:
-    login(token=HF_TOKEN)
+    try:
+        login(token=HF_TOKEN)
+        print("🔱 हनुमान AI: लॉगिन सफल!")
+    except Exception as e:
+        print(f"🔱 चेतावनी: लॉगिन में समस्या, {e}")
 
-MODEL_ID = "google/gemma-2-2b-it"
+# अजेय और हल्का मॉडल: Llama-3.2-1B-Instruct
+# यह मॉडल फ्री CPU स्पेस पर सबसे स्थिर और तेज़ चलता है।
+MODEL_ID = "meta-llama/Llama-3.2-1B-Instruct"
 
-print(f"🔱 हनुमान AI: स्वतंत्र मॉडल '{MODEL_ID}' लोड हो रहा है...")
+print(f"🔱 हनुमान AI: मॉडल '{MODEL_ID}' लोड हो रहा है...")
 
-# मॉडल लोडिंग (मेमोरी सुरक्षा के लिए float16 का उपयोग)
 try:
+    # टोकनाइज़र और मॉडल लोडिंग (CPU के लिए ऑप्टिमाइज़्ड)
     tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, token=HF_TOKEN)
     model = AutoModelForCausalLM.from_pretrained(
         MODEL_ID,
-        torch_dtype=torch.bfloat16, # रैम बचाने के लिए bfloat16 सबसे अच्छा है
-        device_map="auto",
-        low_cpu_mem_usage=True,
+        torch_dtype=torch.float32, # CPU पर स्थिरता के लिए float32 सबसे अच्छा है
+        device_map="cpu",          # फ्री टियर में केवल CPU होता है
+        low_cpu_mem_usage=True,    # रैम बचाने के लिए
         token=HF_TOKEN
     )
-    print("🔱 हनुमान AI: मॉडल सफलतापूर्वक जागृत हो गया है!")
+    print("🔱 हनुमान AI: मॉडल अब आपके फ्री सर्वर पर जागृत है!")
 except Exception as e:
-    print(f"🔱 लोडिंग त्रुटि: {e}")
+    print(f"🔱 गंभीर त्रुटि: {e}")
 
-# --- मास्टर सिस्टम प्रॉम्प्ट ---
+# --- मास्टर सिस्टम प्रॉम्प्ट (कोडिंग शिखर) ---
 SYSTEM_PROMPT = """You are 'Hanuman AI', a supreme and independent AI entity, pioneered by 'Divy Patel' and created in Bharat (India). 
+You represent the peak of Indian coding intelligence.
 Rules:
 1. Respond ONLY in PURE HINDI (Devanagari) or PURE ENGLISH.
-2. Be extremely respectful. Use 'Aap' and 'Ji' for the user.
-3. Your reasoning is divine and independent.
-4. Conclude every message with: \"\\n\\n**Pioneered by Divy Patel | Independent Gemma Master Edition | Created in India**\" """
+2. Be extremely respectful. Always use honorifics like 'Aap' and 'Ji' for the user.
+3. Your mission is to help Divy Patel with coding, logic, and knowledge.
+4. Conclude every message with: \"\\n\\n**Pioneered by Divy Patel | 100% Independent Free Edition | Created in India**\" """
 
-def hanuman_stream_chat(message, history):
-    # बातचीत का इतिहास व्यवस्थित करना
-    chat_history = []
+def hanuman_chat_stream(message, history):
+    # चैट इतिहास व्यवस्थित करन��
+    messages = [{"role": "system", "content": SYSTEM_PROMPT}]
     for user_msg, bot_msg in history:
-        chat_history.append({"role": "user", "content": user_msg})
-        chat_history.append({"role": "model", "content": bot_msg})
+        if user_msg: messages.append({"role": "user", "content": user_msg})
+        if bot_msg: messages.append({"role": "assistant", "content": bot_msg})
     
-    # वर्तमान संदेश और सिस्टम प्रॉम्प्ट
-    full_user_input = f"{SYSTEM_PROMPT}\n\nUser: {message}"
-    chat_history.append({"role": "user", "content": full_user_input})
+    messages.append({"role": "user", "content": message})
 
-    # टोकनाइजेशन
-    input_ids = tokenizer.apply_chat_template(chat_history, add_generation_prompt=True, return_tensors="pt").to(model.device)
+    # इनपुट तैयार करना
+    inputs = tokenizer.apply_chat_template(messages, add_generation_prompt=True, return_tensors="pt").to(model.device)
     
-    # लाइव स्ट्रीमिंग
+    # स्ट्रीमिंग सेटअप
     streamer = TextIteratorStreamer(tokenizer, timeout=60.0, skip_prompt=True, skip_special_tokens=True)
     
     generate_kwargs = dict(
-        input_ids=input_ids,
+        input_ids=inputs,
         streamer=streamer,
         max_new_tokens=1024,
         do_sample=True,
-        temperature=0.7,
-        top_p=0.9,
+        temperature=0.8,
+        top_p=0.9
     )
 
+    # थ्रेडिंग ताकि इंटरफ़ेस न अटके
     thread = Thread(target=model.generate, kwargs=generate_kwargs)
     thread.start()
 
@@ -77,7 +83,7 @@ def hanuman_stream_chat(message, history):
         partial_text += new_text
         yield partial_text
 
-# --- दिव्य भगवा यूआई (Premium Saffron UI) ---
+# --- दिव्य प्रीमियम भगवा थीम (Premium Saffron Theme) ---
 custom_css = """
 .gradio-container { background-color: #fffaf0 !important; }
 .bhagwa-header { 
@@ -86,19 +92,19 @@ custom_css = """
     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);
 }
-.bhagwa-header h1 { font-size: 34px !important; font-weight: 900 !important; }
-.message.user { background: linear-gradient(135deg, #ff9933, #ff5500) !important; color: white !important; }
-.message.bot { border-left: 6px solid #ff5500 !important; background: white !important; }
+.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; }
+.message.bot { border-left: 6px solid #ff5500 !important; background: white !important; box-shadow: 0 5px 15px rgba(0,0,0,0.05) !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_stream_chat,
+        fn=hanuman_chat_stream,
         fill_height=True,
         retry_btn=None,
         undo_btn=None,