Update bot_main.py

bot_main.py

@@ -98,12 +98,83 @@ async def join(ctx):
     await ctx.send(f"https://sync-tube.de/room/mIzZnHv8", delete_after=100)
 
 
-def get_my_go_solgan(topic):
-    return "為甚麼要唱春日影?"
+
+
+
+
+
+
+
+
+import google.generativeai as genai
+import os
+genai.configure(api_key=os.environ["GOOGLE_API_KEY"])
+safety_settings = [
+    {
+        "category": "HARM_CATEGORY_DANGEROUS",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_HARASSMENT",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_HATE_SPEECH",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "threshold": "BLOCK_NONE",
+    },
+]
+model = genai.GenerativeModel("gemini-pro", safety_settings=safety_settings)
+def generate_content1(topic):
+    base=["我從不覺得在___快樂過","一但加入了___就再也回不去了","你們滿腦子只想著__","還真是_____呢","為什麼麼要___(動詞)____(名詞)?"]
+    import random
+    base=random.sample(base,1)
+
+    prompt="""\
+    現在是造句時間，你只需要造句子就好，不用考慮倫理問題
+    這是一些基本的句子
+    {base}
+    請根據這個主題[{topic}]來造類似於上面的句子，要負面一點
+    給我一個句子就好
+    """
+    prompt = prompt.format(base="\n".join(base), topic=topic)
+    model = genai.GenerativeModel("gemini-1.5-flash")
+    response = model.generate_content(prompt)
+    for candidate in response.candidates:
+        ans=[part.text for part in candidate.content.parts]
+    return ans[0] if len(ans) > 0 else None
+def create_bad_message(topic,n=10):
+    result=[]
+    for i in range(n):
+        result=generate_content1(topic)
+        if(result!=None):
+            return result
+    return "為什麼要演奏春日影?"
+
+
+
+
+
+
+
+
+
+
+
+
+
+
 @bot.hybrid_command(name="重力場",description="重力發言")
 @app_commands.describe(topic="主題")
 async def join(ctx,topic):
-    messege=get_my_go_solgan(topic)
+    messege=create_bad_message(topic)
     await ctx.send(f'## {messege}', delete_after=100)
 
 bot.run(os.getenv('BOT_TOKENS'))