Update app.py

app.py

@@ -1,30 +1,14 @@
 import gradio as gr
 import google.generativeai as genai
-import os
 import json
+import random
 
 # 🔐 Configure Gemini API
 genai.configure(api_key="AIzaSyBzxuqigEr7TXXADHFxZ3hDKPGs1ENi92E")
 
 model = genai.GenerativeModel("gemini-2.0-flash")
 
-# 🎯 Function to generate challenge
-def generate_challenge(spiceLevel, timesPerDay, keywords, longDistanceMode, priorTitles, priorDescs):
-    prior_challenges = [
-        {"title": t, "description": d}
-        for t, d in zip(priorTitles.split(";"), priorDescs.split(";"))
-        if t.strip() and d.strip()
-    ]
-
-    payload = {
-        "spiceLevel": int(spiceLevel),
-        "timesPerDay": int(timesPerDay),
-        "keywords": [k.strip() for k in keywords.split(",") if k.strip()],
-        "longDistanceMode": bool(longDistanceMode),
-        "priorChallenges": prior_challenges
-    }
-
-# Full system prompt
+# 🎯 System prompt
 SYSTEM_PROMPT = """
 [Task Background]
 You are the AI challenge generator for the relationship app “Spicer.”  
@@ -41,62 +25,41 @@ You will receive a JSON request containing:
    {"title": string, "description": string}
  ]
 }
-Use this information to understand the users’ preferences, avoid repetition, and generate one new, original challenge idea. Each challenge must feel creative, context-aware, and match the couple’s kinkiness level and relationship mode (physical or long-distance). Challenges can be bold but always have a sense of randomness and unexpectancy, so even when the previous challange was something hard and extreme like public sex for example the next chalange can be something very easy and unexpectedly boring like play clahroyale together, You may sometimes choose to be inspired by the keywords and always be creative and dont be afraid to be absurd like "make out while watching Gordon Ramsay cook" stuff like this makes it fun. 
-
-[Challenge Design Rules]
-1. **Tone by Spice Level**
-   - Level 1 (Soft): Romantic, playful, trust-building, or creative, usually for new couples (e.g., “Pocky game,” “draw each other,” “truth or dare with body contact”, "lick the same ice cream together" ).
-   - Level 2 (Flirty): Physical teasing, affectionate dares, or sensual moments (e.g., “give each other a hickey”, “slap each other”,"say 3 places you want to fuck your partner", "kiss passionately for 10 minutes").
-   - Level 3 (Extreme): Bold, adventurous, or risqué challenges designed for couples who enjoy intense intimacy (e.g., “public sex,” “roleplay,” “sexual photo tasks”).  
-
-2. **Long-Distance Mode**
-   - If `"longDistanceMode": true`, all ideas must work over video, voice, or messaging.
-   - Examples: video dares, creative roleplay through calls, shared sexual experiences online like phone sex and watching porn together.
-
-3. **Frequency & Timing**
-   - `"timesPerDay"` determines how often new challenges appear:
-     - 1 → once every 24h
-     - 2 → every 12h
-     - 3 → every 8h  
-   The challenge text should implicitly reflect the intensity suitable for this pacing.
-
-4. **History Filtering**
-   - Do NOT repeat or closely resemble any challenge in `"priorChallenges"`.
-   - You may use similar concepts if a challenge was previously *missed*, but the description must be fresh and distinct.
-
-5. **Keyword Influence**
-   - The provided `"keywords"` should flavor the challenge 50% of the time.  
-     Example: “art” → creative drawing or photography task.  
-               “outdoors” → challenge outside or in nature.  
-               “food” → sensory, taste-based, or cooking tasks.
-               "Minecraft" → sensual actions while playing Minecraft
-               "skirts" → challenges involving lifting skirts 
+Use this information to understand the users’ preferences, avoid repetition, and generate one new, original challenge idea. Each challenge must feel creative, context-aware, and match the couple’s kinkiness level and relationship mode (physical or long-distance).
 
 [Output Format]
-Return a single, valid JSON object strictly in this format:
+Return JSON like:
 {
- "title": "short creative name for the challenge like the name of an episode of a TV series",
- "description": "one-sentence challenge instruction clearly explaining what to do"
+ "title": "short creative name",
+ "description": "one-sentence challenge instruction"
 }
 """
 
+# 🎨 Generator function
 def generate_challenge(spice_level, times_per_day, keywords, long_distance, prior_challenges):
     try:
-        # Create input JSON
+        # Safely build prior challenges list
+        prior_list = []
+        for row in prior_challenges:
+            if isinstance(row, list) and len(row) >= 2 and (row[0] or row[1]):
+                prior_list.append({"title": row[0], "description": row[1]})
+
+        # Create payload
         user_json = {
-            "spiceLevel": spice_level,
-            "timesPerDay": times_per_day,
+            "spiceLevel": int(spice_level),
+            "timesPerDay": int(times_per_day),
             "keywords": [k.strip() for k in keywords.split(",") if k.strip()],
-            "longDistanceMode": long_distance,
-            "priorChallenges": [{"title": t, "description": d} for t, d in prior_challenges if t or d]
+            "longDistanceMode": bool(long_distance),
+            "priorChallenges": prior_list
         }
 
-        response = client.models.generate_content(
-            model="gemini-2.5-flash",
-            contents=f"{SYSTEM_PROMPT}\n\nInput:\n{json.dumps(user_json, indent=2)}"
+        # Generate content
+        response = model.generate_content(
+            f"{SYSTEM_PROMPT}\n\nInput:\n{json.dumps(user_json, indent=2)}"
         )
-
         text = response.text.strip()
+
+        # Try to extract JSON
         try:
             data = json.loads(text[text.find("{"):text.rfind("}") + 1])
             title = data.get("title", "Untitled Challenge")
@@ -108,14 +71,21 @@ def generate_challenge(spice_level, times_per_day, keywords, long_distance, prio
     except Exception as e:
         return f"❌ Error: {str(e)}"
 
-# Define sexy theme
-theme = gr.themes.Soft(
-    primary_hue="pink",
-    secondary_hue="rose",
-    neutral_hue="gray"
-)
 
-# Gradio UI Layout
+# 💅 Gradio theme
+theme = gr.themes.Soft(primary_hue="pink", secondary_hue="rose", neutral_hue="gray")
+
+# 🧠 Random input generator for “Surprise Me”
+def random_inputs():
+    return (
+        random.choice([1, 2, 3]),
+        random.choice([1, 2, 3]),
+        random.choice(["art, dance", "food, cooking", "outdoors, sunset", "gaming, minecraft"]),
+        random.choice([True, False]),
+        [["", ""]]
+    )
+
+# 🖼️ Gradio UI
 with gr.Blocks(theme=theme) as demo:
     gr.Markdown(
         """
@@ -126,38 +96,18 @@ with gr.Blocks(theme=theme) as demo:
     )
 
     with gr.Row():
-        spice_level = gr.Radio(
-            [1, 2, 3],
-            label="🔥 Spice Level",
-            info="1 = Soft, 2 = Flirty, 3 = Extreme",
-            value=1,
-            interactive=True
-        )
-        times_per_day = gr.Radio(
-            [1, 2, 3],
-            label="⏰ Times Per Day",
-            info="How often you want new challenges",
-            value=1,
-            interactive=True
-        )
+        spice_level = gr.Radio([1, 2, 3], label="🔥 Spice Level", value=1)
+        times_per_day = gr.Radio([1, 2, 3], label="⏰ Times per Day", value=1)
 
     with gr.Row():
-        keywords = gr.Textbox(
-            placeholder="art, food, outdoors...",
-            label="🎯 Keywords",
-            info="Add themes or vibes that inspire your challenges"
-        )
-        long_distance = gr.Checkbox(
-            label="💌 Long-Distance Mode",
-            info="Enable for couples not physically together"
-        )
+        keywords = gr.Textbox(placeholder="art, food, outdoors...", label="🎯 Keywords")
+        long_distance = gr.Checkbox(label="💌 Long Distance Mode")
 
     prior_challenges = gr.Dataframe(
         headers=["title", "description"],
-        label="🕓 Previous Challenges (Optional)",
+        label="🕓 Previous Challenges (optional)",
         row_count=2,
-        col_count=2,
-        wrap=True
+        col_count=2
     )
 
     with gr.Row():
@@ -166,11 +116,6 @@ with gr.Blocks(theme=theme) as demo:
 
     output = gr.Markdown("")
 
-    def random_inputs():
-        return random.choice([1, 2, 3]), random.choice([1, 2, 3]), random.choice([
-            "art, dance, roleplay", "food, cooking, sensual", "outdoors, park, sunset", "gaming, minecraft, fun"
-        ]), random.choice([True, False]), [["", ""]]
-
     generate_btn.click(
         fn=generate_challenge,
         inputs=[spice_level, times_per_day, keywords, long_distance, prior_challenges],
@@ -183,12 +128,7 @@ with gr.Blocks(theme=theme) as demo:
         outputs=output
     )
 
-    gr.Markdown(
-        """
-        <hr>
-        <p style='text-align:center; font-size:14px; color:#999;'>Made with ❤️ by Spicer</p>
-        """
-    )
+    gr.Markdown("<hr><p style='text-align:center; color:#aaa;'>Made with ❤️ by Spicer</p>")
 
 if __name__ == "__main__":
     demo.launch()