more prompts

app.py

@@ -6,10 +6,12 @@ import uuid
 from huggingface_hub import InferenceClient, CommitScheduler, hf_hub_download
 from openai import OpenAI
 from huggingface_hub import get_token, login
-from prompts import detailed_genre_description_prompt, basic_prompt
+from prompts import detailed_genre_description_prompt, basic_prompt, very_basic_prompt
 import random
 import os
 from pathlib import Path
+import random
+from typing import Callable, List, Tuple
 
 HF_TOKEN = os.getenv("HF_TOKEN")
 
@@ -94,13 +96,27 @@ def create_client(model_id):
 # )
 
 
-def generate_prompt():
-    if random.choice([True, False]):
-        return detailed_genre_description_prompt()
-    else:
-        return basic_prompt()
-
 
+def weighted_random_choice(choices: List[Tuple[Callable, float]]) -> Callable:
+    total = sum(weight for _, weight in choices)
+    r = random.uniform(0, total)
+    upto = 0
+    for choice, weight in choices:
+        if upto + weight >= r:
+            return choice
+        upto += weight
+    assert False, "Shouldn't get here"
+
+def generate_prompt() -> str:
+    prompt_choices = [
+        (detailed_genre_description_prompt, 0.4),
+        (basic_prompt, 0.3),
+        (very_basic_prompt, 0.3),
+    ]
+    
+    selected_prompt_func = weighted_random_choice(prompt_choices)
+    return selected_prompt_func()
+    
 def get_and_store_prompt():
     prompt = generate_prompt()
     print(prompt)  # Keep this for debugging