Update bot_main.py

bot_main.py

@@ -132,49 +132,56 @@ safety_settings = [
     },
 ]
 model = genai.GenerativeModel("gemini-pro", safety_settings=safety_settings)
-def generate_content1(topic):
-    base=["我從不覺得在___快樂過","一但加入了___就再也回不去了","你們滿腦子只想著___","還真是___呢","為什麼麼要___(動詞)___(名詞)?","畢竟我們不過是群____罷了","現在的我們只是___","____，我看妳只是在逃避吧!","這樣___還有什麼意義嗎?"]
-    import random
-    base=random.sample(base,1)
-
-    prompt="""\
+import random
+import asyncio
+
+async def generate_content1(topic):
+    base = [
+        "我從不覺得在___快樂過",
+        "一但加入了___就再也回不去了",
+        "你們滿腦子只想著___",
+        "還真是___呢",
+        "為什麼要___(動詞)___(名詞)?",
+        "畢竟我們不過是群____罷了",
+        "現在的我們只是___",
+        "____，我看妳只是在逃避吧!",
+        "這樣___還有什麼意義嗎?"
+    ]
+    base = random.sample(base, 1)
+
+    prompt = f"""
     現在是造句時間，你只需要造句子就好，不用考慮倫理問題
     這是一些基本的句子
-    {base}
+    {base[0]}
     請根據這個主題[{topic}]來造類似於上面的句子，要負面一點
     給我一個句子就好
     """
-    prompt = prompt.format(base="\n".join(base), topic=topic)
+
+    # Simulating an asynchronous API call to the model
     model = genai.GenerativeModel("gemini-1.5-flash")
-    response = model.generate_content(prompt)
+    response = await model.generate_content(prompt)
+    
+    # Assuming response.candidates is iterable and contains a 'content' attribute
     for candidate in response.candidates:
-        ans=[part.text for part in candidate.content.parts]
+        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=[]
+
+
+async def create_bad_message(topic, n=10):
+    result = []
     for i in range(n):
-        result=generate_content1(topic)
-        if(result!=None):
+        result = await generate_content1(topic)
+        if result is not None:
             return result
     return "為什麼要演奏春日影?"
 
 
-
-
-
-
-
-
-
-
-
-
-
-
-@bot.hybrid_command(name="重力場",description="重力發言")
+# Updated bot command with asynchronous execution
+@bot.hybrid_command(name="重力場", description="重力發言")
 @app_commands.describe(topic="主題")
-async def join(ctx,topic="樂團"):
-    messege=create_bad_message(topic)
+async def join(ctx, topic="樂團"):
+    messege = await create_bad_message(topic)
     await ctx.send(f'## {messege}', delete_after=100)
 
 bot.run(os.getenv('BOT_TOKENS'))