Update app.py

app.py

@@ -17,7 +17,7 @@ def generate_text(prompt: str):
         response = genai.generate_text(prompt=prompt)
         result = response.result
     except Exception as e:
-        result = "Something went wrong. Please try again with other prompts."
+        result = "Something went wrong. Please try again with another prompt."
         print(f"Exception {e} occured\n")
     print(f"Bot Message:\n{result}\n")
     return result
@@ -44,9 +44,13 @@ def generate_chat(prompt: str, chat_messages=chat_history):
           messages=chat_messages
         )
         result = response.last
-        chat_messages.append(result)
+        if result is None:
+            result = "Something went wrong. Please try again later or try with another message."
+            chat_messages = chat_messages[:-1]
+        else:
+            chat_messages.append(result)
     except Exception as e:
-        result = "Something went wrong. Please try again with other messages."
+        result = "Something went wrong. Please try again later or try with another message."
         chat_messages = chat_messages[:-1]
         print(f"Exception {e} occured\n")
     chat_history = chat_messages
@@ -62,7 +66,8 @@ with gr.Blocks(theme='JohnSmith9982/small_and_pretty') as demo:
 
     with gr.Tab('Just Text'):
         app = gr.Interface(fn=generate_text, inputs=["text"], outputs=["text"],
-                           examples=[["What is Large Language Model?"], ["Write me a story about Pokemon."]])
+                           # examples=[["What is Large Language Model?"], ["Write me a story about Pokemon."]]
+        )
 
     with gr.Tab('Just Chat'):
         chatbot = gr.Chatbot(height=500)