Update app.py

app.py

@@ -48,20 +48,20 @@ def process_repo_input_and_store(text):
 def show_combined_repo_and_llm():
     global current_repo_idx
     if not last_repo_ids:
-        return "No repo ID available. Please submit repo IDs first.", "", pd.DataFrame(), None
+        return "No repo ID available. Please submit repo IDs first.", "", pd.DataFrame()
     if current_repo_idx >= len(last_repo_ids):
-        return "All repo IDs have been processed.", "", pd.read_csv("repo_ids.csv"), None
+        return "All repo IDs have been processed.", "", pd.read_csv("repo_ids.csv")
     repo_id = last_repo_ids[current_repo_idx]
     try:
         download_space_repo(repo_id, local_dir="repo_files")
     except Exception as e:
-        return f"Error downloading repo: {e}", "", pd.read_csv("repo_ids.csv"), None
+        return f"Error downloading repo: {e}", "", pd.read_csv("repo_ids.csv")
     txt_path = combine_repo_files_for_llm()
     try:
         with open(txt_path, "r", encoding="utf-8") as f:
             combined_content = f.read()
     except Exception as e:
-        return f"Error reading {txt_path}: {e}", "", pd.read_csv("repo_ids.csv"), None
+        return f"Error reading {txt_path}: {e}", "", pd.read_csv("repo_ids.csv")
     llm_output = analyze_combined_file(txt_path)
     llm_json = parse_llm_json_response(llm_output)
     # Update CSV for the current repo id
@@ -81,13 +81,16 @@ def show_combined_repo_and_llm():
                     df.at[idx, "speciality"] = llm_json.get("speciality", "")
                     df.at[idx, "relevance rating"] = llm_json.get("relevance rating", "")
                 break
+        # Add Selected column for highlighting
+        df["Selected"] = ""
+        if highlight_idx is not None:
+            df.at[highlight_idx, "Selected"] = "✅"
         df.to_csv(csv_filename, index=False)
     except Exception as e:
         df = pd.read_csv(csv_filename)
-        highlight_idx = None
     # Move to next repo for next click
     current_repo_idx += 1
-    return combined_content, llm_output, (df, [highlight_idx] if highlight_idx is not None else [])
+    return combined_content, llm_output, df
 
 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"])
@@ -105,8 +108,7 @@ with gr.Blocks() as demo:
     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=True
+        headers=["repo id", "strength", "weaknesses", "speciality", "relevance rating", "Usecase", "Selected"]
     )
     combine_btn.click(show_combined_repo_and_llm, inputs=None, outputs=[combined_txt, llm_output_txt, df_display])