Update app.py

app.py

@@ -2,36 +2,32 @@ import gradio as gr
 from huggingface_hub import InferenceClient
 import random
 
-# Identity & Brand
-NAME = "Infinity"
-DESCRIPTION = "Next-gen Multimodal Reasoning & Creative Engine"
-
-# Connecting to Free Powerful Cloud Models
-# Gemma 4 for Logic/Chat & FLUX for Images
-client_chat = InferenceClient("google/gemma-4-2b-it")
-client_image = InferenceClient("black-forest-labs/FLUX.1-schnell")
+# 1. Multiple Models for Stability (Agar ek fail ho toh dusra chale)
+# Chat ke liye Mistral 7B (Bahut stable hai free tier pe)
+client_chat = InferenceClient("mistralai/Mistral-7B-Instruct-v0.3")
+# Image ke liye Stable Diffusion (FLUX agar busy ho toh ye chalega)
+client_image = InferenceClient("stabilityai/stable-diffusion-xl-base-1.0")
 
 def infinity_engine(message, history):
     user_msg = message.lower()
     
-    # --- STEP 1: IMAGE GENERATION LOGIC ---
-    image_triggers = ["generate", "make", "create", "draw", "photo", "image", "banao", "paint"]
+    # CASE 1: Image Generation
+    image_triggers = ["generate", "make", "create", "draw", "photo", "image", "banao"]
     if any(word in user_msg for word in image_triggers):
-        yield "Infinity is painting your imagination... 🎨"
+        yield "Infinity is painting... 🎨"
         try:
+            # Random seed taaki har baar nayi image bane
             seed = random.randint(0, 1000000)
             img = client_image.text_to_image(message, seed=seed)
             yield img
             return
-        except Exception as e:
-            yield f"Error: {str(e)}. Try again in a moment."
+        except Exception:
+            yield "Infinity: Image server busy hai, 10 second baad phir try karein."
             return
 
-    # --- STEP 2: SMART CHAT LOGIC (Gemma 4) ---
-    system_prompt = f"System: Your name is {NAME}. You are an advanced AI Agent created by RockSky1. You are logical, fast, and multimodal. Never mention Google or Gemma. Identify only as {NAME}."
-    
+    # CASE 2: Text Chat
+    system_prompt = "You are Infinity, a powerful AI by RockSky1. Be cool and smart."
     messages = [{"role": "system", "content": system_prompt}]
-    # Adding history for memory
     for val in history:
         if val[0]: messages.append({"role": "user", "content": val[0]})
         if val[1]: messages.append({"role": "assistant", "content": val[1]})
@@ -39,29 +35,16 @@ def infinity_engine(message, history):
 
     response = ""
     try:
-        for token in client_chat.chat_completion(messages, max_tokens=1000, stream=True):
-            token_str = token.choices[0].delta.content
-            response += token_str
-            yield response
-    except Exception as e:
-        yield "Infinity is currently thinking deeply. Please retry."
+        # Stream=False rakhenge taaki connection break na ho
+        result = client_chat.chat_completion(messages, max_tokens=500)
+        response = result.choices[0].message.content
+        yield response
+    except Exception:
+        yield "Infinity: Connection thoda slow hai, ek baar phir se message bhejo bhai."
 
-# --- STEP 3: PREMIUM UI DESIGN ---
-with gr.Blocks(theme=gr.themes.Soft(primary_hue="indigo", secondary_hue="blue")) as demo:
-    gr.Markdown(f"# ♾️ {NAME} AI")
-    gr.Markdown(f"*{DESCRIPTION}*")
-    
-    chat_ui = gr.ChatInterface(
-        fn=infinity_engine,
-        chatbot=gr.Chatbot(height=500, avatar_images=[None, "https://huggingface.co/front/assets/huggingface_logo-noborder.svg"]),
-        title=f"{NAME} v2.0",
-        examples=[
-            "Who are you?", 
-            "Generate a futuristic 3D avatar of a gamer", 
-            "Write a Python script for a discord bot",
-            "What is the future of Multimodal AI?"
-        ],
-        cache_examples=False,
-    )
+# UI Setup
+with gr.Blocks(theme=gr.themes.Default()) as demo:
+    gr.Markdown("# ♾️ INFINITY AI")
+    gr.ChatInterface(fn=infinity_engine)
 
 demo.launch()