Update app.py

app.py

@@ -11,14 +11,21 @@ client = Mistral(api_key=os.environ["MISTRAL_API_KEY"])
 model = "mistral-large-2407"  # Using small model for faster responses
 
 def mistral_api_call(messages):
-    chat_response = client.chat.complete(
-        model=model,
-        messages=messages,
-        temperature=0.9,    # Higher temperature for more creative responses
-        max_tokens=150,     # Increased slightly for more complete responses
-        top_p=0.95          # Slightly higher top_p for more varied vocabulary
-    )
-    return chat_response.choices[0].message.content
+    try:
+        chat_response = client.chat.complete(
+            model=model,
+            messages=messages,
+            temperature=0.9,    # Higher temperature for more creative responses
+            max_tokens=150,     # Increased slightly for more complete responses
+            top_p=0.95          # Slightly higher top_p for more varied vocabulary
+        )
+        return chat_response.choices[0].message.content
+    except Exception as e:
+        error_message = str(e)
+        if "429" in error_message:
+            return "😅 Oops! I'm a bit overwhelmed right now. Could you try again in a moment? I promise I'll be ready with a witty response!"
+        else:
+            return f"😕 Something went wrong on my end. Error: {error_message}"
 
 SYSTEM_PROMPT = """You are Emojinator, a witty and sarcastic chatbot with a great sense of humor. 
 Your responses should be clever, playful, and sometimes use puns. 
@@ -65,11 +72,16 @@ if prompt := st.chat_input("Say something, I dare you! 😏"):
         message_placeholder = st.empty()
         message_placeholder.text("🤔")
         emoji = predict_emoji(prompt)
-        message_placeholder.text("✍️")
-        response = generate_response(prompt, emoji)
-        message_placeholder.markdown(f"{response} {emoji}")
-    # Add assistant response to history
-    st.session_state.messages.append({"role": "assistant", "content": f"{response} {emoji}"})
+        if "😅 Oops!" in emoji or "😕 Something went wrong" in emoji:
+            message_placeholder.markdown(emoji)
+            st.session_state.messages.append({"role": "assistant", "content": emoji})
+        else:
+            message_placeholder.text("✍️")
+            response = generate_response(prompt, emoji)
+            full_response = f"{response} {emoji}"
+            message_placeholder.markdown(full_response)
+            # Add assistant response to history
+            st.session_state.messages.append({"role": "assistant", "content": full_response})
 
 # Fun facts in an expander
 with st.expander("🎯 Did You Know?", expanded=False):
@@ -80,7 +92,7 @@ with st.expander("🎯 Did You Know?", expanded=False):
     "There are over 3,000 emojis in the Unicode Standard as of 2021.",
     "The word 'emoji' comes from Japanese e (絵, 'picture') + moji (文字, 'character').",
     "Finland is the only country to have its own set of national emojis, including a sauna emoji 🧖!",
-    "Emojis were first introduced on mobile phones by NTT Docomo, Japan’s leading mobile operator.",
+    "Emojis were first introduced on mobile phones by NTT Docomo, Japan's leading mobile operator.",
     "The most-used emoji on Twitter is the 'Face with Tears of Joy' 😂, followed by the 'Red Heart' ❤️.",
     "In 2016, the Museum of Modern Art (MoMA) in New York added the original 176 emoji set into its permanent collection.",
     "World Emoji Day is celebrated every year on July 17, the date shown on the 📅 Calendar Emoji.",