Update app.py

app.py

@@ -32,20 +32,25 @@ def compute_final_suggested_location(rows):
     return counts, (top_location, count)
 
 # Store feedback (with required fields)
-def store_feedback_to_drive(accession, answer1, answer2, contact=""):
+import gspread
+from oauth2client.service_account import ServiceAccountCredentials
+
+def store_feedback_to_google_sheets(accession, answer1, answer2, contact=""):
     if not answer1.strip() or not answer2.strip():
         return "⚠️ Please answer both questions before submitting."
 
-    feedback_file = "data/user_fb/feedback_mtdna.csv"
-    header = ["accession", "helpful", "improvement", "contact"]
-    row = [accession, answer1, answer2, contact]
-    file_exists = os.path.isfile(feedback_file)
-    with open(feedback_file, "a", newline="") as f:
-        writer = csv.writer(f)
-        #if not file_exists:
-        writer.writerow(header)
-        writer.writerow(row)
-    return "✅ Feedback submitted. Thank you!"
+    try:
+        # Define the scope and authenticate
+        scope = ["https://spreadsheets.google.com/feeds", "https://www.googleapis.com/auth/drive"]
+        creds = ServiceAccountCredentials.from_json_keyfile_name("credentials.json", scope)
+        client = gspread.authorize(creds)
+
+        # Open the spreadsheet and worksheet
+        sheet = client.open("feedback_mtdna").sheet1  # You can change the name
+        sheet.append_row([accession, answer1, answer2, contact])
+        return "✅ Feedback submitted. Thank you!"
+    except Exception as e:
+        return f"❌ Error submitting feedback: {str(e)}"
 
 def summarize_results(accession):
     try:
@@ -128,7 +133,7 @@ with gr.Blocks() as interface:
 
     run_button.click(fn=classify_with_loading, inputs=accession, outputs=status)
     run_button.click(fn=classify_main, inputs=accession, outputs=[output_table, output_summary, status])
-    submit_feedback.click(fn=store_feedback_to_drive, inputs=[accession, q1, q2, contact], outputs=feedback_status)
+    submit_feedback.click(fn=store_feedback_to_google_sheets, inputs=[accession, q1, q2, contact], outputs=feedback_status)
     reset_button.click(fn=reset_fields, inputs=[], outputs=[accession, q1, q2, contact, feedback_status, output_table, output_summary, status])
 
 interface.launch(share=True)