Update app.py

app.py

@@ -23,17 +23,79 @@ credentials = Credentials.from_service_account_info(service_account_info, scopes
 service = build('sheets', 'v4', credentials=credentials)
 sheet = service.spreadsheets()
 
-# Save user information to Google Sheets
+def fetch_user_details(email):
+    """
+    Fetch user details from Google Sheet based on email.
+    """
+    try:
+        result = sheet.values().get(spreadsheetId=SPREADSHEET_ID, range="Sheet1!A:G").execute()
+        rows = result.get('values', [])
+        for row in rows:
+            if row[0] == email:  # Assuming email is in the first column (A)
+                return {
+                    "email": row[0],
+                    "name": row[1],
+                    "professional_title": row[2],
+                    "industry": row[3],
+                    "target_audience": row[4],
+                    "personal_background": row[5]
+                    "company_url": row[6]
+                }
+        return None
+    except Exception as e:
+        return f"Error fetching details: {str(e)}"
+
 def save_user_info(email, name, professional_title, industry, target_audience, personal_background, company_url):
-    data = [[email, name, professional_title, industry, target_audience, personal_background, company_url]]
-    sheet.values().append(
-        spreadsheetId=SPREADSHEET_ID,
-        range="Sheet1!A2",  # Assuming headers are in row 1
-        valueInputOption="RAW",
-        insertDataOption="INSERT_ROWS",
-        body={"values": data}
-    ).execute()
-    return "User information saved successfully!"
+    """
+    Save or update user details in Google Sheet.
+    """
+    try:
+        result = sheet.values().get(spreadsheetId=SPREADSHEET_ID, range="Sheet1!A:G").execute()
+        rows = result.get('values', [])
+        updated = False
+
+        # Check if email exists and update the row
+        for i, row in enumerate(rows):
+            if row[0] == email:
+                rows[i] = [email, name, professional_title, industry, target_audience, personal_background, company_url]
+                updated = True
+                break
+
+        if not updated:  # Append a new row if email doesn't exist
+            rows.append([email, name, professional_title, industry, target_audience, personal_background, company_url])
+
+        # Save updated rows back to the sheet
+        body = {"values": rows}
+        sheet.values().update(
+            spreadsheetId=SPREADSHEET_ID, range="Sheet1!A:G",
+            valueInputOption="RAW", body=body
+        ).execute()
+        return "Details saved successfully."
+    except Exception as e:
+        return f"Error saving details: {str(e)}"
+
+
+
+def handle_user_details(email, name=None, professional_title=None, industry=None, target_audience=None, personal_background=None, company_url=None, action="verify"):
+    if action == "verify":
+        # Fetch details based on email
+        user_details = fetch_user_details(email)
+        if user_details:
+            return (
+                f"Record found for {email}. You can update details if needed.",
+                user_details  # Return existing details
+            )
+        else:
+            return (
+                f"No record found for {email}. Please enter details and save.",
+                {}  # Return empty fields
+            )
+
+    elif action == "save":
+        # Save or update details
+        result = save_user_info(email, name, professional_title, industry, target_audience, personal_background, company_url)
+        return result, {}
+
 
 # Step 1: User profile management UI
 def user_profile(email, name, professional_title, industry, target_audience, personal_background, company_url):
@@ -130,22 +192,86 @@ def generate_email(prospect_name, linkedin_url, website_url, context_url, word_c
 def email_agent(prospect_name, linkedin_url, website_url, context_url, word_count, email_purpose, interested_position):
     return "<div style='font-style: italic;'>Hold on tight, your personalized email is on the way...</div>", generate_email(prospect_name, linkedin_url, website_url, context_url, word_count, email_purpose, interested_position)
 
-# Gradio Interfaces
+# Gradio Interface
+def verify_user(email):
+    # Logic for verifying the email
+    message, user_details = handle_user_details(email, action="verify")
+    # Populate fields if user details are found
+    if user_details:
+        return (
+            message,
+            gr.update(visible=True, value=user_details.get("name", "")),
+            gr.update(visible=True, value=user_details.get("professional_title", "")),
+            gr.update(visible=True, value=user_details.get("industry", "")),
+            gr.update(visible=True, value=user_details.get("target_audience", "")),
+            gr.update(visible=True, value=user_details.get("personal_background", "")),
+            gr.update(visible=True, value=user_details.get("company_url", ""))
+        )
+    else:
+        # Show empty fields for new user
+        return (
+            message,
+            gr.update(visible=True, value=""),
+            gr.update(visible=True, value=""),
+            gr.update(visible=True, value=""),
+            gr.update(visible=True, value=""),
+            gr.update(visible=True, value=""),
+            gr.update(visible=True, value="")
+        )
+
+
+def save_user(email, name, professional_title, industry, target_audience, personal_background, company_url):
+    # Logic for saving the user details
+    message, _ = handle_user_details(email, name, professional_title, industry, target_audience, personal_background, company_url, action="save")
+    return message
+
 profile_iface = gr.Interface(
-    fn=user_profile,
-    inputs=[
-        gr.Textbox(label="Your Email ID"),
-        gr.Textbox(label="Your Name"),
-        gr.Textbox(label="Professional Title"),
-        gr.Textbox(label="Industry"),
-        gr.Textbox(label="Target Audience"),
-        gr.Textbox(label="Personal Background"),
-        gr.Textbox(label="Company URL"),
-    ],
-    outputs="text",
-    title="Step 1: User Profile",
-    description="Enter and save your personal and company information."
-)
+with gr.Blocks() as ui:
+    with gr.Row():
+        email_input = gr.Textbox(label="Email", placeholder="Enter your email")
+        verify_button = gr.Button("Verify")
+        status = gr.Textbox(label="Status", interactive=False)
+
+    # Other fields are initially hidden
+    with gr.Row(visible=False) as details_section:
+        name_input = gr.Textbox(label="Name")
+        professional_title_input = gr.Textbox(label="Professional Title")
+        industry_input = gr.Textbox(label="Industry")
+        target_audience_input = gr.Textbox(label="Target Audience")
+        personal_background_input = gr.Textbox(label="Personal Background")
+        company_url_input = gr.Textbox(label="Company URL")
+        save_button = gr.Button("Save")
+
+    # Events
+    verify_button.click(
+        verify_user,
+        inputs=[email_input],
+        outputs=[
+            status,
+            name_input,
+            professional_title_input,
+            industry_input,
+            target_audience_input,
+            personal_background_input,
+            company_url_input,
+        ]
+        title="Step 1: User Profile",
+        description="Enter and save your personal and company information."
+    )
+
+    save_button.click(
+        save_user,
+        inputs=[
+            email_input,
+            name_input,
+            professional_title_input,
+            industry_input,
+            target_audience_input,
+            personal_background_input,
+            company_url_input,
+        ],
+        outputs=[status]
+    )
 
 email_iface = gr.Interface(
     fn=email_agent,