Create app.py

app.py

@@ -0,0 +1,84 @@
+import gradio as gr
+import os
+import base64
+import requests
+import json
+from groq import Groq
+
+# Load Groq API key from environment (Gradio Spaces -> Secrets)
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+if not GROQ_API_KEY:
+    raise ValueError("GROQ_API_KEY not found. Please add it as a secret in your Hugging Face Space.")
+
+client = Groq(api_key=GROQ_API_KEY)
+
+# Function to handle image upload and Groq API call
+def generate_catalog(image):
+    if image is None:
+        return "", "", "", ""
+
+    # Convert uploaded image to base64
+    mime_type = "image/jpeg"
+    with open(image.name, "rb") as f:
+        encoded_image = base64.b64encode(f.read()).decode("utf-8")
+    image_data = f"data:{mime_type};base64,{encoded_image}"
+
+    # Prompt
+    prompt = (
+        "You are an expert e-commerce assistant. Given a product image, generate the following in JSON format:\n"
+        "- A short, catchy Title (max 8 words)\n"
+        "- A full Description (100–150 words)\n"
+        "- 5–7 SEO Keywords\n"
+        "- 5–7 relevant Tags\n\n"
+        "Output only in JSON with keys: title, description, keywords, tags."
+    )
+
+    # Groq API call
+    try:
+        response = client.chat.completions.create(
+            model="meta-llama/llama-4-scout-17b-16e-instruct",
+            messages=[
+                {
+                    "role": "user",
+                    "content": [
+                        {"type": "text", "text": prompt},
+                        {"type": "image_url", "image_url": {"url": image_data}},
+                    ],
+                }
+            ],
+            response_format={"type": "json_object"},
+            max_completion_tokens=1024,
+            temperature=0.5,
+            top_p=1,
+            stream=False,
+        )
+
+        parsed = json.loads(response.choices[0].message.content)
+        title = parsed.get("title", "")
+        description = parsed.get("description", "")
+        keywords = ", ".join(parsed.get("keywords", []))
+        tags = ", ".join(parsed.get("tags", []))
+        return title, description, keywords, tags
+
+    except Exception as e:
+        return "Error", str(e), "", ""
+
+# Gradio UI
+with gr.Blocks(theme=gr.themes.Soft(primary_hue="purple")) as app:
+    gr.Markdown("""
+    # 🛒 Instant Product Catalog Assistant
+    Upload a product image and instantly get a title, description, keywords, and tags!
+    """)
+
+    with gr.Row():
+        image_input = gr.File(label="Upload Product Image", type="filepath")
+        submit_btn = gr.Button("Generate Catalog")
+
+    output_title = gr.Textbox(label="Title")
+    output_description = gr.Textbox(label="Description", lines=5)
+    output_keywords = gr.Textbox(label="Keywords (comma separated)")
+    output_tags = gr.Textbox(label="Tags (comma separated)")
+
+    submit_btn.click(generate_catalog, inputs=[image_input], outputs=[output_title, output_description, output_keywords, output_tags])
+
+app.launch()