Update app.py

app.py

@@ -26,16 +26,13 @@ for msg in msgs.messages:
     st.chat_message(msg.type).write(msg.content)
 
 if prompt := st.chat_input():
-    try:
-        st.chat_message("human").write(prompt)
-        msgs.add_user_message(prompt)
-    
-        with st.spinner("Waiting for response..."):
-            res = get_prompt(prompt, api)
-        if res:
-            st.chat_message("ai").write(res)
-            msgs.add_ai_message(res)
-        else:
-            st.error("No valid response received from the AI.")
-    except:
-        return "Something went wrong (:"
+    st.chat_message("human").write(prompt)
+    msgs.add_user_message(prompt)
+
+    with st.spinner("Waiting for response..."):
+        res = get_prompt(prompt, api)
+    if res:
+        st.chat_message("ai").write(res)
+        msgs.add_ai_message(res)
+    else:
+        st.error("No valid response received from the AI.")