Update app.py

app.py

@@ -4,7 +4,7 @@ import requests
 @st.cache_data
 def get_top_relevant_normalized_job_titles(job_title, num_answers):
     # JAMES API endpoint for the top-N relevant normalized job titles
-    api_endpoint = f'http://james-mapping.xyz/api/v1/related?job_title={job_title}'
+    api_endpoint = f'http://james-mapping.xyz/api/v1/related?job_title={job_title}&num_answers={num_answers}'
 
     try:
         # Send GET request to JAMES API
@@ -18,7 +18,17 @@ def get_top_relevant_normalized_job_titles(job_title, num_answers):
             if not normed_job_title_list:
                 return ["No normalized titles found"]
 
-            return normed_job_title_list  # Remove slicing to return all results
+            # Ensure that you have at least 10 results
+            while len(normed_job_title_list) < 10:
+                # Fetch more data until you have at least 10 results
+                num_answers += 1
+                api_endpoint = f'http://james-mapping.xyz/api/v1/related?job_title={job_title}&num_answers={num_answers}'
+                response = requests.get(api_endpoint)
+                response_data = response.json()
+                new_titles = response_data.get("normed_job_title_list", [])
+                normed_job_title_list.extend(new_titles)
+
+            return normed_job_title_list
         else:
             # API call unsuccessful
             return [f'Error: {response.status_code}']
@@ -40,7 +50,7 @@ def main():
         top_normalized_titles = get_top_relevant_normalized_job_titles(job_title, num_answers)
 
         st.subheader(f'Top {num_answers} Relevant Titles:')
-        for i, normed_job_title in enumerate(top_normalized_titles, 1):
+        for i, normed_job_title in enumerate(top_normalized_titles[:num_answers], 1):
             st.text(f'{i}: {normed_job_title}')
 
 if __name__ == '__main__':