Fixed minor bugs

app.py

@@ -42,7 +42,7 @@ if uploaded_file is not None:
         mime_type = "application/pdf"
     else:
         st.error("Unsupported file type. Please upload an image or PDF.")
-        return
+        st.stop()
 
     # Reset chat history when a new file is uploaded
     reset_chat()
@@ -60,21 +60,22 @@ if uploaded_file is not None:
             for message in st.session_state.messages:
                 with st.chat_message(message["role"]):
                     st.markdown(message["content"])
-
-            # Upload the file with the correct MIME type
-            file_data = genai.upload_file(uploaded_file, mime_type=mime_type)
-
-            # Send file and prompt to Gemini API
-            response = model.generate_content(
-                [
-                    user_input, 
-                    file_data
-                ]
-            )
-
-            # Add Gemini response to chat history
-            st.session_state.messages.append({"role": "assistant", "content": response.text})
-
-            # Display Gemini response
-            with st.chat_message("assistant"):
-                st.markdown(response.text)
+            
+            with st.spinner("Processing..."):
+                # Upload the file with the correct MIME type
+                file_data = genai.upload_file(uploaded_file, mime_type=mime_type)
+
+                # Send file and prompt to Gemini API
+                response = model.generate_content(
+                    [
+                        user_input, 
+                        file_data
+                    ]
+                )
+
+                # Add Gemini response to chat history
+                st.session_state.messages.append({"role": "assistant", "content": response.text})
+
+                # Display Gemini response
+                with st.chat_message("assistant"):
+                    st.markdown(response.text)