Upload folder using huggingface_hub

app.py

@@ -1,141 +1,138 @@
 import gradio as gr
 import os
 from openai import OpenAI
+from huggingface_hub import InferenceClient
 from ebooklib import epub
 import datetime
+import requests
+from PIL import Image
+import io
 
 # ==========================================
-# 🧠 AGENT LOGIC: SHARED
+# 🧠 AGENT LOGIC: UPGRADED COVER ART AGENT
 # ==========================================
 
-api_key = os.getenv("OPENAI_API_KEY")
-client = OpenAI(api_key=api_key) if api_key else None
-
-# ==========================================
-# 🏪 SHOPFRONT DATA
-# ==========================================
-
-INVENTORY = [
-    {
-        "title": "AI Passive Income: 2026",
-        "price": "$29.99",
-        "desc": "Launch a 6-figure digital empire using autonomous AI agents. The blueprint for the agentic revolution.",
-        "img": "https://img.freepik.com/free-vector/blue-futuristic-networking-technology-background_53876-120614.jpg",
-        "niche": "Passive Income"
-    },
-    {
-        "title": "Micro-SaaS Masterclass",
-        "price": "$49.99",
-        "desc": "Build lean, high-profit software with AI. From concept to $10k MRR in 30 days.",
-        "img": "https://img.freepik.com/free-vector/digital-technology-blueprint-background_53876-114441.jpg",
-        "niche": "SaaS"
-    },
-    {
-        "title": "The Zero-Waste Home",
-        "price": "$19.99",
-        "desc": "Save $5,000 a year by optimizing your life with smart automation. Sustainability for beginners.",
-        "img": "https://img.freepik.com/free-vector/eco-friendly-house-technology-background_53876-115343.jpg",
-        "niche": "Sustainability"
-    }
-]
+class CoverArtAgent:
+    def __init__(self, openai_client, hf_token):
+        self.openai_client = openai_client
+        self.hf_client = InferenceClient(token=hf_token) if hf_token else None
+        self.styles = {
+            "Cinematic": "highly detailed, cinematic lighting, sharp focus, 8k, photorealistic",
+            "Minimalist": "clean lines, negative space, modern typography, flat design, vector aesthetic",
+            "Cyberpunk": "neon lights, rainy streets, high contrast, futuristic, glow effects",
+            "Oil Painting": "textured brushstrokes, classical composition, rich colors, fine art",
+            "Blueprint": "technical drawing, blueprint grid, architectural, precise, blueprint blue"
+        }
+
+    def art_director_synthesis(self, title, subtitle, summary, style):
+        """
+        Uses GPT to act as an Art Director and generate a pro-level prompt.
+        """
+        if not self.openai_client:
+            return f"Book cover for {title}. Style: {style}"
+        
+        try:
+            response = self.openai_client.chat.completions.create(
+                model="gpt-4o-mini",
+                messages=[{
+                    "role": "system", 
+                    "content": "You are a world-class book cover art director. Your task is to generate a highly descriptive, professional prompt for an image generation AI."
+                }, {
+                    "role": "user",
+                    "content": f"Book Title: {title}\nSubtitle: {subtitle}\nSummary: {summary}\nStyle: {style}\n\nGenerate a 100-word detailed visual prompt for DALL-E 3 or Flux.1."
+                }]
+            )
+            return response.choices[0].message.content
+        except:
+            return f"A professional book cover for '{title}' in {style} style, representing {summary[:50]}."
+
+    def generate(self, model_choice, title, subtitle, summary, style_choice):
+        # 1. Art Direction Phase
+        pro_prompt = self.art_director_synthesis(title, subtitle, summary, style_choice)
+        
+        # 2. Render Phase
+        if model_choice == "DALL-E 3":
+            if not self.openai_client: return None, "OpenAI Key Missing."
+            try:
+                res = self.openai_client.images.generate(model="dall-e-3", prompt=pro_prompt, quality="hd")
+                return res.data[0].url, f"Art Director Prompt: {pro_prompt}"
+            except Exception as e: return None, str(e)
+            
+        elif model_choice == "Flux.1 (Hugging Face)":
+            if not self.hf_client: return None, "HF Token Missing (Check Space Settings)."
+            try:
+                # Use Flux.1-dev or Schnell via HF Inference
+                image = self.hf_client.text_to_image(pro_prompt, model="black-forest-labs/FLUX.1-dev")
+                return image, f"Art Director Prompt: {pro_prompt}"
+            except Exception as e: return None, str(e)
 
 # ==========================================
-# ⚙️ AGENT CLASSES
+# 💰 AGENT LOGIC: FINANCE
 # ==========================================
 
-class IntegratedFactoryOrchestrator:
-    def __init__(self, agents): self.agents = agents
-    def run_complete_build(self, title, subtitle, author, niche, summary):
-        now = datetime.datetime.now().strftime("%H:%M:%S")
-        logs = [f"[{now}] 🚀 STARTING FULL BUILD: {title}"]
-        img, _ = self.agents['cover'].analyze_and_generate(title, subtitle, summary)
-        logs.append(f"[SUCCESS] Build Pipeline Finished.")
-        return "\n".join(logs), img, "$1,450.00 Est. Profit", "Originality: 99% (Approved)"
-
 class FinanceAgent:
     def audit_and_forecast(self, title, niche, price):
-        return f"--- FINANCE REPORT: {title} ---\n[*] COGS: $0.50\n[*] Est. Monthly Profit: $1,450.00"
-
-class LegalGuardAgent:
-    def audit_manuscript(self, title, author, text, niche):
-        return "Audit: Approved.", f"© 2026 {author}. All Rights Reserved."
-
-class EpubGeneratorAgent:
-    def create_epub(self, title, author, content, path):
-        return f"Built: {path}"
+        return f"--- FINANCE REPORT: {title} ---\n[*] Est. Monthly Profit: $1,450.00"
 
-class CoverArtAgent:
-    def __init__(self, client): self.client = client
-    def analyze_and_generate(self, title, subtitle, summary):
-        if not self.client: return "https://via.placeholder.com/300x400.png?text=Cover+Pending", "Strategy: Placeholder"
-        try:
-            res = self.client.images.generate(model="dall-e-3", prompt=f"Professional book cover for {title}. {summary[:50]}")
-            return res.data[0].url, "DALL-E 3 Success."
-        except: return "https://via.placeholder.com/300x400.png?text=API+Error", "Error."
+# ==========================================
+# 📅 AGENT LOGIC: CALENDAR
+# ==========================================
 
-class MarketingSalesAgent:
-    def generate_growth_package(self, title, niche):
-        return f"🚀 GROWTH PACKAGE: {title} for {niche}."
+class CalendarOrchestratorAgent:
+    def get_schedule(self):
+        today = datetime.date.today()
+        return [{"Date": str(today + datetime.timedelta(days=i)), "Book": b, "Status": "⏳ Scheduled"} for i, b in enumerate(["Modern Parenting", "The Un-Touristed Path"])]
 
 # ==========================================
 # 🤖 SHARED STATE
 # ==========================================
 
-agents = {
-    'finance': FinanceAgent(),
-    'legal': LegalGuardAgent(),
-    'epub': EpubGeneratorAgent(),
-    'cover': CoverArtAgent(client),
-    'marketing': MarketingSalesAgent()
-}
-orchestrator = IntegratedFactoryOrchestrator(agents)
+api_key = os.getenv("OPENAI_API_KEY")
+hf_token = os.getenv("HF_TOKEN") # Pull from Space secrets
+o_client = OpenAI(api_key=api_key) if api_key else None
+
+cover_agent = CoverArtAgent(o_client, hf_token)
+finance_agent = FinanceAgent()
+calendar_agent = CalendarOrchestratorAgent()
 
 # ==========================================
 # 🎨 UI DEFINITION
 # ==========================================
 
 with gr.Blocks(title="📚 EbookBuilder Studio", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 📚 EbookBuilder Studio & Shopfront")
+    gr.Markdown("# 📚 EbookBuilder Studio: Upgraded Edition")
     
     with gr.Tabs():
-        with gr.TabItem("🛒 Digital Shopfront"):
-            gr.Markdown("## 💎 Premium AI-Generated Trilogy")
-            gr.Markdown("Explore our flagship collection of autonomous guides for the 2026 economy.")
-            
-            for item in INVENTORY:
-                with gr.Row():
-                    with gr.Column(scale=1):
-                        gr.Image(value=item['img'], label=item['title'], interactive=False)
-                    with gr.Column(scale=2):
-                        gr.Markdown(f"### {item['title']}")
-                        gr.Markdown(f"**Price: {item['price']}** | *Niche: {item['niche']}*")
-                        gr.Markdown(f"{item['desc']}")
-                        buy_btn = gr.Button(f"🛒 Purchase {item['title']}", variant="primary")
-                gr.Markdown("---")
-
-        with gr.TabItem("⚡ Factory (Test Run)"):
+        with gr.TabItem("🎨 Advanced Image Gen"):
+            gr.Markdown("### 🛠 Multi-Model Cover Art Agent")
             with gr.Row():
                 with gr.Column():
-                    t = gr.Textbox(label="Title", value="Modern Parenting")
-                    st = gr.Textbox(label="Subtitle", value="Resilient Kids in the Age of AI")
-                    a = gr.Textbox(label="Author", value="EbookBuilder AI")
-                    n = gr.Textbox(label="Niche", value="Modern Parenting")
-                    s = gr.Textbox(label="Summary", value="A guide for digital-first families.", lines=3)
-                    run_btn = gr.Button("🚀 Execute Full Build Sequence", variant="primary")
+                    m_choice = gr.Radio(["DALL-E 3", "Flux.1 (Hugging Face)"], label="Select Rendering Engine", value="DALL-E 3")
+                    s_choice = gr.Dropdown(list(cover_agent.styles.keys()), label="Artistic Style", value="Cinematic")
+                    ct = gr.Textbox(label="Book Title", placeholder="e.g. AI Passive Income")
+                    cst = gr.Textbox(label="Subtitle")
+                    csum = gr.Textbox(label="Summary", lines=3)
+                    gen_btn = gr.Button("🚀 Execute Multi-Agent Generation", variant="primary")
                 with gr.Column():
-                    factory_logs = gr.Textbox(label="Factory Pipeline Logs", lines=10)
-                    res_img = gr.Image(label="Generated Cover Art")
-
-            run_btn.click(fn=orchestrator.run_complete_build, inputs=[t, st, a, n, s], outputs=[factory_logs, res_img, gr.State(), gr.State()])
+                    c_out = gr.Image(label="Final Render")
+                    c_log = gr.Textbox(label="Art Director's Visual Strategy", lines=8)
+            
+            gen_btn.click(fn=cover_agent.generate, inputs=[m_choice, ct, cst, csum, s_choice], outputs=[c_out, c_log])
 
-        with gr.TabItem("💰 Finance"):
-            gr.Markdown("### 💹 ROI & Profitability Dashboard")
+        with gr.TabItem("🛒 Shopfront"):
+            gr.Markdown("## 💎 Premium AI-Generated Trilogy")
             with gr.Row():
-                with gr.Column():
-                    ft = gr.Textbox(label="Title"); fn = gr.Dropdown(choices=["Passive Income", "SaaS", "Sustainability"], label="Niche"); fp = gr.Number(label="Price ($)", value=29.99)
-                    calc_btn = gr.Button("📊 Run Audit")
-                with gr.Column(): f_out = gr.Textbox(label="Report", lines=10)
-            calc_btn.click(fn=agents['finance'].audit_and_forecast, inputs=[ft, fn, fp], outputs=[f_out])
+                gr.Image("https://img.freepik.com/free-vector/blue-futuristic-networking-technology-background_53876-120614.jpg", label="AI Passive Income")
+                gr.Image("https://img.freepik.com/free-vector/digital-technology-blueprint-background_53876-114441.jpg", label="Micro-SaaS Masterclass")
+            
+        with gr.TabItem("📅 Factory"):
+            gr.Dataframe(value=calendar_agent.get_schedule())
+            
+        with gr.TabItem("💰 Finance"):
+            gr.Markdown("### 💹 ROI Dashboard")
+            ft = gr.Textbox(label="Title"); fn = gr.Dropdown(["Passive Income", "SaaS"], label="Niche"); fp = gr.Number(label="Price", value=29.99)
+            f_btn = gr.Button("Run Audit")
+            f_btn.click(fn=finance_agent.audit_and_forecast, inputs=[ft, fn, fp], outputs=[gr.Textbox()])
 
 if __name__ == "__main__":
     demo.launch()

requirements.txt

@@ -3,3 +3,4 @@ gradio
 ebooklib
 beautifulsoup4
 stripe
+huggingface_hub