Update app.py

app.py

@@ -232,12 +232,44 @@ def interface():
             dropdown.change(lambda c, ci, l: (1,), inputs=[country_filter, city_filter, language_filter], outputs=[page_state])
             dropdown.change(filter_by_fields, inputs=[country_filter, city_filter, language_filter, page_state], outputs=[table_html, page_state, gr.Number(visible=False)])
 
-        prev_btn.click(lambda p: max(p - 1, 1), inputs=[page_state], outputs=[page_state])
-        prev_btn.click(filter_by_fields, inputs=[country_filter, city_filter, language_filter, page_state], outputs=[table_html, page_state, gr.Number(visible=False)])
-
-        next_btn.click(lambda p, c, ci, l: min((filter_by_fields(c, ci, l, p + 1)[2] - 1) // PAGE_SIZE + 1, p + 1),
-                       inputs=[page_state, country_filter, city_filter, language_filter], outputs=[page_state])
-        next_btn.click(filter_by_fields, inputs=[country_filter, city_filter, language_filter, page_state], outputs=[table_html, page_state, gr.Number(visible=False)])
+        def get_total_pages(selected_country, selected_city, selected_language):
+            filters = []
+            params = []
+            if selected_country != "All":
+                filters.append("country = ?")
+                params.append(selected_country.title())
+            if selected_city != "All":
+                filters.append("city = ?")
+                params.append(selected_city.title())
+            if selected_language != "All":
+                filters.append("languages LIKE ?")
+                params.append(f"%{selected_language.lower()}%")
+            query = "SELECT COUNT(*) FROM teamup"
+            if filters:
+                query += " WHERE " + " AND ".join(filters)
+            with sqlite3.connect(DB_PATH) as conn:
+                total = pd.read_sql_query(query, conn, params=params).iloc[0, 0]
+            return max(1, (total - 1) // PAGE_SIZE + 1)
+
+        prev_btn.click(
+            fn=lambda p: max(p - 1, 1),
+            inputs=[page_state],
+            outputs=[page_state]
+        ).then(
+            fn=filter_by_fields,
+            inputs=[country_filter, city_filter, language_filter, page_state],
+            outputs=[table_html, page_state, gr.Number(visible=False)]
+        )
+
+        next_btn.click(
+            fn=lambda p, c, ci, l: min(get_total_pages(c, ci, l), p + 1),
+            inputs=[page_state, country_filter, city_filter, language_filter],
+            outputs=[page_state]
+        ).then(
+            fn=filter_by_fields,
+            inputs=[country_filter, city_filter, language_filter, page_state],
+            outputs=[table_html, page_state, gr.Number(visible=False)]
+        )
 
 
         del_btn = gr.Button("Delete Profile")