Update bot_main.py

bot_main.py

@@ -147,6 +147,7 @@ def generate_content1(topic):
         "____，我看妳只是在逃避吧!",
         "這樣___還有什麼意義嗎?"
     ]
+    
     base = random.sample(base, 1)
 
     prompt = f"""
@@ -156,14 +157,17 @@ def generate_content1(topic):
     請根據這個主題[{topic}]來造類似於上面的句子，要負面一點
     給我一個句子就好
     """
-
-    # Simulating an asynchronous API call to the model
-    model = genai.GenerativeModel("gemini-1.5-flash")
-    response = model.generate_content(prompt)
+    try:
+        model = genai.GenerativeModel("gemini-1.5-flash")
+        response = 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]
+    except error:
+        ans=[]
     
-    # Assuming response.candidates is iterable and contains a 'content' attribute
-    for candidate in response.candidates:
-        ans = [part.text for part in candidate.content.parts]
+   
     
     return ans[0] if len(ans) > 0 else None
 
@@ -171,7 +175,9 @@ def generate_content1(topic):
 async def create_bad_message(topic, n=10):
     result = []
     for i in range(n):
+        print("gen",i)
         result = generate_content1(topic)
+        print("end_gen",i)
         if result is not None:
             return result
     return "為什麼要演奏春日影?"