Update app.py

app.py

@@ -6,6 +6,7 @@ import os
 import uuid
 import requests
 import replicate
+from groq import Groq
 
 # ============================
 #  CONFIG
@@ -13,6 +14,10 @@ import replicate
 REPLICATE_API_TOKEN = "r8_OHfjZvEbVoU9fuioi4qfQ50OHdpAmqV4OAa5b"
 os.environ["REPLICATE_API_TOKEN"] = REPLICATE_API_TOKEN
 
+# === MASUKKAN API KEY GROQ DI SINI ===
+GROQ_API_KEY = "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"
+client = Groq(api_key=GROQ_API_KEY)
+
 MAX_CHAPTERS = 10
 
 # ============================
@@ -42,20 +47,16 @@ def ensure_outputs_dir():
     os.makedirs("outputs", exist_ok=True)
 
 # ============================
-#  AI TEXT GENERATOR (LLaMA‑3 FIXED)
+#  AI TEXT GENERATOR (GROQ LLaMA‑3 70B)
 # ============================
 def ai_generate(prompt):
     try:
-        output = replicate.run(
-            "meta/meta-llama-3-8b-instruct",
-            input={
-                "messages": [
-                    {"role": "user", "content": prompt}
-                ],
-                "max_tokens": 500
-            }
+        response = client.chat.completions.create(
+            model="llama3-70b-8192",
+            messages=[{"role": "user", "content": prompt}],
+            max_tokens=600
         )
-        return "".join(output)
+        return response.choices[0].message.content
     except Exception as e:
         return f"AI gagal menghasilkan teks. Error: {str(e)}"
 
@@ -195,7 +196,7 @@ def build_ebook_pdf_with_images(username, goal, genre, tone, style, length,
             pdf.image(illus_path, x=20, y=y_start, w=170)
             pdf.set_y(y_start + 80)
 
-        # AI CHAPTER TEXT
+        # AI CHAPTER TEXT (GROQ)
         chapter_text = generate_chapter_text(goal, style, tone, ch)
 
         pdf.set_font("Helvetica", "", 12)
@@ -271,7 +272,7 @@ with gr.Blocks() as demo:
 
     with gr.Group(visible=False) as main_ui:
 
-        gr.Markdown("# 🌟 AIPromptLab — JSON & PDF E-Book Generator (Replicate FREE Models)")
+        gr.Markdown("# 🌟 AIPromptLab — JSON & PDF E-Book Generator (Groq + Replicate)")
 
         with gr.Tabs():