Update app.py

app.py

@@ -2,54 +2,42 @@ import os, io, time, requests, gradio as gr
 from PIL import Image
 import google.generativeai as genai
 
-# Load secure keys
 HF_TOKEN = os.getenv("HF_API_TOKEN")
 GEMINI_KEY = os.getenv("GEMINI_API_KEY")
 
-# Configure Gemini
 genai.configure(api_key=GEMINI_KEY)
-GEMINI_MODEL = "gemini-pro"
 
-# HF Image Model
-HF_URL = "https://api-inference.huggingface.co/models/prompthero/openjourney"
-HEADERS = {"Authorization": f"Bearer {HF_TOKEN}"}
+def chat_fn(message, history):
+    if not history:
+        history = []
 
-# Chatbot
-def chat(query, history):
-    history = history or []
     try:
-        result = genai.GenerativeModel(GEMINI_MODEL).generate_content(query)
-        reply = result.text
+        model = genai.GenerativeModel("gemini-pro")
+        response = model.generate_content(message)
+        answer = response.text
     except Exception as e:
-        reply = f"Error: {e}"
-    history.append((query, reply))
-    return history, history
+        answer = f"Error: {e}"
+
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": answer})
+
+    # Display format for Gradio Chatbot UI
+    display = [(h["content"] if h["role"]=="user" else None,
+                h["content"] if h["role"]=="assistant" else None)
+               for h in history if h["role"] in ["user","assistant"]]
+
+    return display, history
+
+HF_IMAGE_MODEL = "stabilityai/stable-diffusion-2-1"
+HF_URL = f"https://api-inference.huggingface.co/models/{HF_IMAGE_MODEL}"
+HDRS = {"Authorization": f"Bearer {HF_TOKEN}"}
 
-# Image Generation
 def generate_image(prompt):
-    if not prompt:
-        return None
     payload = {"inputs": prompt, "options": {"wait_for_model": True}}
-    r = requests.post(HF_URL, headers=HEADERS, json=payload)
-    if r.status_code != 200:
-        return None
+    r = requests.post(HF_URL, headers=HDRS, json=payload)
+
     img = Image.open(io.BytesIO(r.content)).convert("RGB")
     return img
 
-# UI
 with gr.Blocks() as app:
-    gr.Markdown("<h1 style='text-align:center;'>⚔ Ronin AI Studio</h1>")
-
-    with gr.Tab("💬 Chat"):
-        chatbox = gr.Chatbot()
-        msg = gr.Textbox(label="Ask me anything")
-        send = gr.Button("Send")
-        send.click(chat, [msg, chatbox], [chatbox, chatbox])
-
-    with gr.Tab("🎨 Text to Image"):
-        prompt = gr.Textbox(label="Describe an image")
-        btn = gr.Button("Generate")
-        output = gr.Image(type="pil")
-        btn.click(generate_image, prompt, output)
-
-app.launch()
+    gr.Markdown("<h1 style='text-align:center