Update app.py

app.py

@@ -26,18 +26,16 @@ for msg in msgs.messages:
     st.chat_message(msg.type).write(msg.content)
 
 if prompt := st.chat_input():
-    start_time = time.time()
-    st.chat_message("human").write(prompt)
-    msgs.add_user_message(prompt)
-
-    with st.spinner("Waiting for response..."):
-        try:
+    try:
+        st.chat_message("human").write(prompt)
+        msgs.add_user_message(prompt)
+    
+        with st.spinner("Waiting for response..."):
             res = get_prompt(prompt, api)
-        except Exception as e:
-            return "Something went wrong (:"
-    if res:
-        st.chat_message("ai").write(res)
-        end_time = time.time()
-        msgs.add_ai_message(res)
-    else:
-        st.error("No valid response received from the AI.")
+        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 (:"