Update app.py

app.py

@@ -69,6 +69,9 @@ def show_combined_repo_and_llm():
     try:
         df = pd.read_csv(csv_filename)
         highlight_idx = None
+        # Cast columns to string to avoid dtype issues
+        for col in ["strength", "weaknesses", "speciality", "relevance rating"]:
+            df[col] = df[col].astype(str)
         for idx, row in df.iterrows():
             if row["repo id"] == repo_id:
                 highlight_idx = idx
@@ -84,7 +87,7 @@ def show_combined_repo_and_llm():
         highlight_idx = None
     # Move to next repo for next click
     current_repo_idx += 1
-    return combined_content, llm_output, df, highlight_idx
+    return combined_content, llm_output, (df, [highlight_idx] if highlight_idx is not None else [])
 
 repo_id_input = gr.Textbox(label="Enter repo IDs (comma or newline separated)", lines=5, placeholder="repo1, repo2\nrepo3")
 df_output = gr.Dataframe(headers=["repo id", "strength", "weaknesses", "speciality", "relevance rating", "Usecase"])
@@ -101,8 +104,10 @@ with gr.Blocks() as demo:
     combine_btn = gr.Button("Download, Combine & Show .py/.md Files from Next Repo and Analyze")
     combined_txt = gr.Textbox(label="Combined Repo Files", lines=20)
     llm_output_txt = gr.Textbox(label="LLM Analysis Output", lines=10)
-    df_display = gr.Dataframe(headers=["repo id", "strength", "weaknesses", "speciality", "relevance rating", "Usecase"])
-    highlight_row = gr.HighlightedText(label="Highlighted Row Index")
-    combine_btn.click(show_combined_repo_and_llm, inputs=None, outputs=[combined_txt, llm_output_txt, df_display, highlight_row])
+    df_display = gr.Dataframe(
+        headers=["repo id", "strength", "weaknesses", "speciality", "relevance rating", "Usecase"],
+        highlight_row=True
+    )
+    combine_btn.click(show_combined_repo_and_llm, inputs=None, outputs=[combined_txt, llm_output_txt, df_display])
 
 demo.launch()