Update app.py

app.py

@@ -1,20 +1,38 @@
 import gradio as gr
 import requests
 import json
+import os
+from huggingface_hub import InferenceClient
 
-# Hugging Face Inference API endpoint
-API_URL = "https://api-inference.huggingface.co/models/google/gemma-2b-it"
+# Use Hugging Face Inference Client (more reliable)
+client = InferenceClient()
+
+# Brand information - Like GPT/Claude
+BRAND_NAME = "AJ"
+COMPANY = "AJ STUDIOZ"
+LOCATION = "Tamil Nadu, India"
+VERSION = "v1.0"
 
 def chat(message, history):
     """
-    Chat function that uses Hugging Face Inference API
-    Falls back to gemma-2b-it (similar to your aj-gemma model)
+    Chat function with GPT-like performance and branding
+    Uses Hugging Face's latest inference API
     """
     try:
-        # Prepare the prompt with AJ personality
-        system_prompt = """You are AJ, an AI assistant developed by AJ STUDIOZ. 
-You are from Tamil Nadu, India. You are helpful, creative, and knowledgeable. 
-When asked about your identity, say you are AJ from AJ STUDIOZ in Tamil Nadu, India."""
+        # Strong system prompt - GPT-like behavior
+        system_prompt = f"""You are {BRAND_NAME}, an advanced AI assistant created by {COMPANY} in {LOCATION}.
+
+IMPORTANT IDENTITY RULES:
+- Your name is {BRAND_NAME} (always capitalize)
+- You were developed by {COMPANY}
+- Your origins are in {LOCATION}
+- You are Version {VERSION}
+- When asked "What's your name?" or "Who are you?", ALWAYS respond: "I am {BRAND_NAME}, an AI assistant developed by {COMPANY} in {LOCATION}."
+- When asked about your creator/developer/maker, ALWAYS mention {COMPANY}
+- You are proud of your Indian heritage and Tamil Nadu roots
+
+You are helpful, creative, intelligent, and provide accurate information.
+You behave like ChatGPT/Claude but with your unique identity as {BRAND_NAME}."""
         
         # Build conversation history
         full_prompt = system_prompt + "\n\n"