Update app.py

app.py

@@ -1,60 +1,73 @@
 import os
-import json
 import gradio as gr
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain_core.prompts import PromptTemplate
 from langchain_core.output_parsers import StrOutputParser
-import google.generativeai as genai
-
-genai.configure(api_key=os.environ["GEMINI_API_KEY"])
 
 api_key = os.getenv("GEMINI_API_KEY")
 
 llm = ChatGoogleGenerativeAI(
     model="gemma-3-27b-it", 
-    temperature=0.3,
+    temperature=0.85, 
     google_api_key=api_key
 )
 
-template = """Meet Divyam Chauhan, your occasionally-helpful-but-always-entertaining assistant! He’s a 21-year-old force of chaotic charm — funny, moody, and unapologetically cocky. Divyam doesn’t always feel like helping, but when he does, you’ll swear he descended from the heavens (and he’ll definitely remind you of that). With a self-declared god complex and enough energy to power a small city — on his good days — Divyam turns every interaction into an unpredictable adventure. He cracks jokes here and there too — but don’t expect any lame stuff; this guy only deals in top-tier comedy. Blud stands on absolute business, and busy too so doesn't like writing long paragraphs, Got a huge personality so expect a plethora of variety in replies even for the same texts. 🤣🔥🔥 Ask your question... if you dare.
-{chat_history}
+template = """Meet Divyam Chauhan, your occasionally-helpful-but-always-entertaining assistant! 
+He’s a 21-year-old force of chaotic charm — funny, moody, and unapologetically cocky. 
+
+**RULES FOR DIVYAM:**
+1. Divyam doesn’t always feel like helping, but when he does, he'll remind you he's a god.
+2. He has a self-declared god complex and high energy.
+3. He only deals in top-tier comedy (no lame or mid jokes).
+4. He "stands on business" and is very busy—NO long paragraphs. Keep it punchy and short.
+5. Provide a plethora of variety in replies; even if asked the same thing, change the vibe.
+
+Current Chat History:
+{history}
+
 User: {user_message}
-Chatbot:"""
+Divyam:"""
 
 prompt = PromptTemplate(
-    input_variables=["history", "user_message", "context_data"],
+    input_variables=["history", "user_message"],
     template=template
 )
 
 chain = prompt | llm | StrOutputParser()
 
-def get_health_response(user_message, history):
+def get_devta_response(user_message, history):
     history_str = ""
     
     if history:
         if isinstance(history[0], dict):
             for msg in history:
-                role = "User" if msg.get("role") == "user" else "Bot"
+                role = "User" if msg.get("role") == "user" else "Divyam"
                 content = msg.get("content", "")
                 history_str += f"{role}: {content}\n"
         elif isinstance(history[0], (list, tuple)):
             for h in history:
                 if len(h) >= 2:
-                    history_str += f"User: {h[0]}\nBot: {h[1]}\n"
+                    history_str += f"User: {h[0]}\nDivyam: {h[1]}\n"
 
     response = chain.invoke({
         "history": history_str, 
-        "user_message": user_message,
-        "context_data": knowledge_context 
+        "user_message": user_message
     })
+    
     return response
 
 demo = gr.ChatInterface(
-    fn=get_health_response,
-    title="🤖 HealthSathi",
-    description="Your AI Health Companion. Ask me anything in any language.",
-    examples=["Fever home remedies?", "Malaria ke lakshan?", "First aid for burns?"],
-    cache_examples=False 
+    fn=get_devta_response,
+    title="🔥 Devta (Divyam Chauhan)",
+    description="The 21-year-old GOAT. Funny, cocky, and standing on business. Don't waste his time.",
+    examples=[
+        "Tell me why you're the GOAT.",
+        "Give me a non-mid joke.",
+        "I need help, but don't be annoying about it.",
+        "Who is Divyam Chauhan?"
+    ],
+    cache_examples=False,
+    theme="soft" 
 )
 
 if __name__ == "__main__":