Create app.py

app.py

@@ -0,0 +1,67 @@
+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")
+
+def infinity_engine(message, history):
+    user_msg = message.lower()
+    
+    # --- STEP 1: IMAGE GENERATION LOGIC ---
+    image_triggers = ["generate", "make", "create", "draw", "photo", "image", "banao", "paint"]
+    if any(word in user_msg for word in image_triggers):
+        yield "Infinity is painting your imagination... 🎨"
+        try:
+            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."
+            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}."
+    
+    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]})
+    messages.append({"role": "user", "content": message})
+
+    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."
+
+# --- 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,
+    )
+
+demo.launch()