Refactor trivia and weather functions; remove trivia-related code and update help text

app.py

@@ -101,7 +101,6 @@ client = BotClient(BotConfig)
 # --- Threading, Queues, Stores ---
 
 task_queue   = queue.Queue()
-trivia_store = {}
 polls        = {}
 last_message_time = time.time()
 
@@ -169,48 +168,10 @@ def _fn_weather(message_id, chat_id, loc):
         "prompt":f"Speak only this weather report: {report}"
     })
 
-def _fn_weather_poem(message_id, chat_id, loc):
-    raw = requests.get(f"http://sl.wttr.in/{loc}?format=4", timeout=5).text
-    poem = generate_llm(
-        f"Write a short, poetic weather summary in Celsius based on:\n\n{raw}"
-    )
-    client.send_message(message_id, chat_id, poem)
-    task_queue.put({
-        "type":"audio","message_id":message_id,"chat_id":chat_id,
-        "prompt":f"Speak only this poetic weather summary: {poem}"
-    })
-
 def _fn_inspire(message_id, chat_id):
     quote = generate_llm(f"Give me a short inspirational unique quote.")
     client.send_message(message_id, chat_id, f"✨ {quote}")
 
-def _fn_trivia(message_id, chat_id):
-    raw = generate_llm(
-        f"Generate a unique trivia Q&A in JSON: {{\"question\":\"...\",\"answer\":\"...\"}}"
-    )
-    try:
-        obj = json.loads(raw.strip().strip("```json").strip("```"))
-        trivia_store[chat_id] = obj
-        client.send_message(
-            message_id, chat_id,
-            f"❓ {obj['question']}\nReply `/answer` or `/answer your guess`."
-        )
-    except:
-        client.send_message(message_id, chat_id, "Failed to generate trivia.")
-
-def _fn_answer(message_id, chat_id, guess):
-    if chat_id not in trivia_store:
-        client.send_message(message_id, chat_id, "No active trivia. `/trivia` to start.")
-        return
-    qa = trivia_store.pop(chat_id)
-    if guess:
-        verdict = generate_llm(
-            f"Q: {qa['question']}\nCorrect: {qa['answer']}\nUser: {guess}\nCorrect?"
-        )
-        client.send_message(message_id, chat_id, verdict)
-    else:
-        client.send_message(message_id, chat_id, f"💡 Answer: {qa['answer']}")
-
 def _fn_meme(message_id, chat_id, txt):
     client.send_message(message_id, chat_id, "🎨 Generating your meme...")
     task_queue.put({"type":"image","message_id":message_id,
@@ -334,9 +295,7 @@ help_text = (
     "• /translate <lang>|<text>\n"
     "• /joke\n"
     "• /weather <loc>\n"
-    "• /weatherpoem <loc>\n"
     "• /inspire\n"
-    "• /trivia / /answer\n"
     "• /meme <text>\n"
     "• /poll <Q>|<opt1>|… / /results / /endpoll\n"
     "• /gen <prompt>|<count>\n"
@@ -403,14 +362,8 @@ async def whatsapp_webhook(request: Request):
         _fn_joke(mid, chat_id); return {"success": True}
     if low.startswith("/weather "):
         _fn_weather(mid, chat_id, body[9:].strip().replace(" ","+")); return {"success": True}
-    if low.startswith("/weatherpoem "):
-        _fn_weather_poem(mid, chat_id, body[13:].strip().replace(" ","+")); return {"success": True}
     if low == "/inspire":
         _fn_inspire(mid, chat_id); return {"success": True}
-    if low == "/trivia":
-        _fn_trivia(mid, chat_id); return {"success": True}
-    if low.startswith("/answer"):
-        _fn_answer(mid, chat_id, body[7:].strip()); return {"success": True}
     if low.startswith("/meme "):
         _fn_meme(mid, chat_id, body[6:].strip()); return {"success": True}
     if low.startswith("/poll "):

config.yaml

@@ -10,10 +10,7 @@ config:
         • /translate <lang>|<text> — translate text
         • /joke — tell a short joke
         • /weather <location> — short, creative weather report in °C
-        • /weatherpoem <location> — poetic weather summary in °C
         • /inspire — inspirational quote
-        • /trivia — start a trivia question
-        • /answer — answer the current trivia
         • /meme <text> — generate a meme image
         • /poll <Q>|<opt1>|<opt2>|… — create a poll
         • /results — show poll results

polLLM.py

@@ -1,12 +1,16 @@
 import os
-import requests
-import urllib.parse
+from openai import OpenAI
 from dotenv import load_dotenv
 from utils import read_config
 import random
 
 load_dotenv()
 
+client = OpenAI(
+    base_url="https://text.pollinations.ai/openai",
+    api_key="YOUR_API_KEY"  # Add if needed
+)
+
 def pre_process():
     # Read the configuration and substitute the character placeholder
     config = read_config()
@@ -14,25 +18,27 @@ def pre_process():
     char = config['llm']['char']
     return system_prompt.replace("{char}", char)
 
-def generate_llm(prompt):
+def generate_llm(prompt, model="openai-large", max_tokens=100):
     system_prompt = pre_process()
-    # Encode the user prompt and system prompt for URL safety
-    encoded_prompt = urllib.parse.quote(prompt)
-    encoded_system = urllib.parse.quote(system_prompt)
-    # Build the GET request URL for Pollinations' text API
-    randomSeed = random.randint(0, 9999999)
-    print(f"DEBUG: Random seed: {randomSeed}")
-    url = f"https://text.pollinations.ai/{encoded_prompt}?model=openai-large&private=true&system={encoded_system}&seed={randomSeed}"
     
     try:
-        response = requests.get(url, timeout=30)
-        response.raise_for_status()
-        # Return the generated text (stripping any extra whitespace)
-        return response.text.strip()
+        # Use OpenAI's ChatCompletion API
+        randomSeed = random.randint(0, 9999999)
+        response = client.chat.completions.create(
+            model=model,
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": prompt}
+            ],
+            max_tokens=max_tokens,
+            seed=randomSeed
+        )
+        # Return the generated text
+        return response.choices[0].message.content.strip()
     except Exception as e:
         return f"Error: {str(e)}"
 
 # Example usage (can be removed or commented out in production):
 if __name__ == "__main__":
-    sample_prompt = "What is the capital of France?"
-    print("Response:", generate_llm(sample_prompt))
+    sample_prompt = f"Generate a unique trivia Q&A in JSON: {{\"question\":\"...\",\"answer\":\"...\"}}"
+    print("Response:", generate_llm(sample_prompt))