Update app.py

app.py

@@ -182,7 +182,7 @@ def analyze_competitors(row, co):
     query = row['query']
     our_url = row['page']
     
-    competitor_urls = get_serp_results(query)
+    competitor_urls = get_serp_results(query)[:5]  # Get top 5 competitors
     
     results = []
     
@@ -270,7 +270,7 @@ def analyze_competitors(row, co):
     our_url = row['page']
     our_score = row['relevancy_score']
     
-    competitor_urls = get_serp_results(query)
+    competitor_urls = get_serp_results(query)[:5]  # Get top 5 competitors
     
     results = []
     for url in competitor_urls:
@@ -498,12 +498,31 @@ def show_tabular_data(df, co):
         score = calculate_relevance_score(content, row['query'], co)
         return score
 
+    # Function to analyze competitors
+    def analyze_competitors(row):
+        query = row['query']
+        our_url = row['page']
+        our_score = row['relevancy_score']
+        
+        competitor_urls = get_serp_results(query)[:5]  # Get top 5 competitors
+        
+        results = []
+        for url in competitor_urls:
+            content = fetch_content(url)
+            score = calculate_relevance_score(content, query, co)
+            results.append({'url': url, 'relevancy_score': score})
+        
+        results.append({'url': our_url, 'relevancy_score': our_score})
+        results_df = pd.DataFrame(results).sort_values('relevancy_score', ascending=False)
+        
+        return results_df
+
     # Display the dataframe
     st.dataframe(df_display)
 
     # Create buttons for each row
     for index, row in df_display.iterrows():
-        col1, col2 = st.columns([3, 1])
+        col1, col2, col3 = st.columns([3, 1, 1])
         with col1:
             st.write(f"Query: {row['query']} | Page: {row['page']}")
         with col2:
@@ -516,13 +535,11 @@ def show_tabular_data(df, co):
                     st.experimental_rerun()
             else:
                 st.write(f"Relevancy Score: {row['relevancy_score']:.4f}")
-
-    # Add an expander for each row to show competitor analysis
-    for index, row in df.iterrows():
-        with st.expander(f"Analyze Competitors for: {row['query']} | {row['page']}"):
-            if st.button("Analyze Competitors", key=f"comp_{index}"):
+        with col3:
+            if st.button(f"Compare Competitors", key=f"comp_{index}"):
                 with st.spinner('Analyzing competitors...'):
-                    results_df = analyze_competitors(row, co)
+                    results_df = analyze_competitors(row)
+                    st.write("Competitor Comparison:")
                     st.dataframe(results_df)
                     
                     our_rank = results_df.index[results_df['url'] == row['page']].tolist()[0] + 1