Update app.py

app.py

@@ -1,93 +1,57 @@
 import gradio as gr
 import pandas as pd
 
-# Load your data (replace with your actual file path if local or URL if remote)
-df = pd.read_excel("your_file.xlsx")
+def process_google_sheet(sheet_url):
+    # Convert Google Sheets URL to CSV export
+    if "/edit" in sheet_url:
+        sheet_url = sheet_url.replace("/edit", "/export?format=csv")
+    elif not sheet_url.endswith("/export?format=csv"):
+        sheet_url += "/export?format=csv"
+
+    try:
+        df = pd.read_csv(sheet_url)
+    except Exception as e:
+        return f"<span style='color:red'>Error reading sheet: {e}</span>"
 
-# Mapping for AG column
-mapping = {
-    "obs_commercial_purpose_commercial": "Commercial",
-    "obs_commercial_purpose_issue": "SIP",
-    "obs_environmental_politics_global": "Environmental Politics",
-    "obs_civil_social_rights_global": "Civil and Social Rights",
-    "obs_economy_global": "Economy",
-    "obs_security_foreign_policy_global": "Security and Foreign Policy",
-    "obs_political_values_global": "Political Values and Governance",
-    "obs_health_global": "Health",
-    "obs_immigration_global": "Immigration",
-    "obs_crime_global": "Crime",
-    "obs_guns_global": "Guns",
-    "obs_education_global": "Education"
-}
-
-def process_row(name, job_id, col_r, col_af, col_ag, col_cc):
-    # Handle TM (col_r)
-    tm_value = mapping.get(col_r, "NT") if pd.notna(col_r) else "NT"
-    
-    # Handle Auditor (col_af + col_ag)
-    auditor_tags = []
-    if pd.notna(col_af):
-        auditor_tags.append(mapping.get(col_af, "NT"))
-    else:
-        auditor_tags.append("NT")
-    if pd.notna(col_ag):
-        if col_ag == "obs_commercial_purpose_commercial":
-            auditor_tags = [mapping[col_ag]]  # only "Commercial"
-        else:
-            auditor_tags.append(mapping.get(col_ag, ""))
-    auditor_value = ", ".join([x for x in auditor_tags if x])
-
-    # Special rule: SIP always at end
-    if "SIP" in auditor_tags and len(auditor_tags) > 1:
-        auditor_tags = [x for x in auditor_tags if x != "SIP"] + ["SIP"]
-        auditor_value = ", ".join(auditor_tags)
-
-    # Highlighting logic
-    if tm_value == auditor_value:
-        tm_display = f'<span style="color:green;font-weight:bold">{tm_value}</span>'
-        auditor_display = f'<span style="color:green;font-weight:bold">{auditor_value}</span>'
-    else:
-        tm_display = f'<span style="color:red;font-weight:bold">{tm_value}</span>'
-        auditor_display = f'<span style="color:red;font-weight:bold">{auditor_value}</span>'
-
-    # Rationale (col_cc)
-    rationale = col_cc if pd.notna(col_cc) else ""
-
-    # Format final output
-    output = (
-        f"{name}<br>"
-        f"job id: #{job_id}<br>"
-        f"TM: {tm_display}<br>"
-        f"Auditor: {auditor_display}<br><br>"
-        f"Rationale: {rationale}"
-    )
-    return output
-
-def generate_output(selected_name):
-    person_data = df[df["BX"] == selected_name]  # column BX = Names
     results = []
-    for _, row in person_data.iterrows():
-        result = process_row(
-            row["BX"],   # Name
-            row["A"],    # Job ID
-            row["R"],    # TM column
-            row["AF"],   # Auditor column
-            row["AG"],   # Auditor issue/commercial
-            row["CC"]    # Rationale
-        )
+    for _, row in df.iterrows():
+        name = row.get("Name", "")
+        job_id = row.get("Job ID", "")
+
+        tm_answer = str(row.get("R", "NT")).strip() if pd.notna(row.get("R")) else "NT"
+        auditor_answer = str(row.get("AF", "NT")).strip() if pd.notna(row.get("AF")) else "NT"
+        rationale = str(row.get("CC", "")).strip()
+
+        # Compare TM vs Auditor
+        if tm_answer == auditor_answer:
+            tm_display = f"<span style='color:green;font-weight:bold'>{tm_answer}</span>"
+            auditor_display = f"<span style='color:green;font-weight:bold'>{auditor_answer}</span>"
+        else:
+            tm_display = f"<span style='color:red;font-weight:bold'>{tm_answer}</span>"
+            auditor_display = f"<span style='color:red;font-weight:bold'>{auditor_answer}</span>"
+
+        result = f"""
+        <div style='margin-bottom:20px'>
+            <b>{name}</b><br>
+            job id: #{job_id}<br>
+            TM: {tm_display}<br>
+            Auditor: {auditor_display}<br><br>
+            Rationale: {rationale}
+        </div>
+        """
         results.append(result)
-    return "<br><br>".join(results)
 
-# Gradio Interface
-names = df["BX"].dropna().unique().tolist()
+    return "<br>".join(results)
+
 
 with gr.Blocks() as demo:
-    gr.Markdown("### Job Review Tool")
+    gr.Markdown("## Google Sheets Report Formatter with Highlighted Comparison")
+
     with gr.Row():
-        name_dropdown = gr.Dropdown(choices=names, label="Select Name")
-    output_html = gr.HTML()
+        sheet_url = gr.Textbox(label="Paste Google Sheet Link")
+
+    output = gr.HTML()
 
-    name_dropdown.change(fn=generate_output, inputs=name_dropdown, outputs=output_html)
+    sheet_url.change(process_google_sheet, inputs=sheet_url, outputs=output)
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()