Update app.py

app.py

@@ -3,316 +3,260 @@ import gspread
 from google.oauth2.service_account import Credentials
 import os
 import json
-from datetime import datetime
-import math
+from threading import Lock
 
 conf1 = os.getenv('CONF')
 dict = json.loads(conf1)
+# Google Sheets settings
 SHEET_ID = os.getenv('SHEET_ID')
 
 SCOPES = ['https://www.googleapis.com/auth/spreadsheets', 'https://www.googleapis.com/auth/drive']
 
+# Gradio login credentials
+log = os.getenv('LOGIN')
+password = os.getenv('PASSWORD')
+
 # Initialize Google Sheets client
 creds = Credentials.from_service_account_info(dict, scopes=SCOPES)
 gc = gspread.authorize(creds)
 sheet = gc.open_by_key(SHEET_ID)
 
-# Track active user sessions and pagination state
-active_users = {}
-page_states = {}
+# Create a lock for synchronizing sheet access
+sheet_lock = Lock()
 
-def calculate_rfm_score(latest_meeting, visit_times, call_times, monetary_value):
-    days_since = 999
-    try:
-        # Calculate recency score
-        last_meeting_date = datetime.strptime(latest_meeting, '%Y-%m-%d')
-        days_since = (datetime.now() - last_meeting_date).days
-
-        # If meeting was within last 5 days, give low priority (they were just visited)
-        if days_since < 5:
-            recency_score = 1
-        # For sleeping customers (no contact in 3+ months), give higher priority
-        elif days_since > 30:
-            recency_score = 5  # High priority to wake them up
-        else:
-            # Normal recency scoring for others
-            recency_score = 5 - min(4, math.floor(days_since / 30))  # 0-4 months scale
-    except:
-        recency_score = 1
-
-    # Calculate frequency score (1 visit = 5 calls)
-    visit_equivalent = int(visit_times) * 5 if visit_times else 0
-    call_equivalent = int(call_times) if call_times else 0
-    total_interactions = visit_equivalent + call_equivalent
-
-    # Base frequency score
-    frequency_score = min(5, math.ceil(total_interactions / 5))  # Every 5 interactions = 1 point, max 5
-
-    # Bonus for consistent interaction (high number of total interactions)
-    if total_interactions > 20:  # Indicates a close partner
-        frequency_score = min(5, frequency_score + 1)  # Bonus point for long-term relationship
-
-    # Monetary score is directly input by user (1-5 scale)
-    monetary_score = min(5, max(1, int(monetary_value))) if monetary_value else 1
-
-    # Calculate total RFM score with adjusted weights
-    # More weight on recency for sleeping customers and monetary value for close partners
-    if days_since > 30:  # Sleeping customer
-        total_score = (recency_score * 0.4 + frequency_score * 0.3 + monetary_score * 0.3) * 20
-    elif total_interactions > 20:  # Close partner
-        total_score = (recency_score * 0.3 + frequency_score * 0.4 + monetary_score * 0.3) * 20
-    else:  # Normal customer
-        total_score = (recency_score * 0.35 + frequency_score * 0.35 + monetary_score * 0.3) * 20
-
-    return round(total_score)
+# Dictionary to store customer info per user session
+user_customers = {}
 
 def verify_username(username):
     """Verify if username exists in user form"""
+    telescripts = '''Hello, this is (your name ) from CEV auto, a professional car export company.
+
+the reason i called you today is to introduce you to our services if you have a minute?
+
+as i told you we are a professional car export company,We have built a platform that specifically helps businesses source vehicles from China and streamline the entire purchasing and export process.
+
+Have you purchased Chinese vehicles before? If so, which brands or models?
+("If not: "Many companies like yours are now exploring Chinese vehicles for their competitiveness and affordability.")
+To better understand your business, could you tell me which country you're based in and the types of products your company focuses on?"
+
+(if they are not in the car business you say:) think about it, its an untapped market, where few people get to make a lot of money out of it, however this business is getting popular,for us we are not looking for just clients, we are looking for potential partners who can join us to make a movement.
+lets arrange a time we meet and get to know each other more, and you get to know more about our business and team. i will add you on wechat to  fix a time.
+
+--------------------------------------------------------------------------------------------------------
+
+why we call: two reasons:
+1-to get the client to come to our office to buy.
+2-to get informations about them ( country, what products do they sell)
+--------------------------------------------------------------------------------------------------------
+
+Introduction:
+"Hello [Customer's Name], this is [Your Name] from CEVauto, your trusted partner in automotive exports. How are you doing today?"
+
+Acknowledge Previous Contact:
+"I'm reaching out because we noticed you've shown interest in our services before, and we truly appreciate your consideration of CEVauto for your automotive needs."
+
+Value Proposition:
+"We wanted to let you know about some exciting updates that could benefit your business. At CEVauto, we've been working hard to enhance our offerings, and now we're proud to provide even more competitive pricing and more stable delivery timelines than ever before."
+
+Benefits:
+"By choosing CEVauto, you can expect:
+
+Cost-effective solutions that maximize your budget.
+
+Reliable delivery schedules to keep your operations running smoothly.
+
+A wide range of high-quality vehicles to meet your specific requirements."
+
+Call to Action:
+"We believe these improvements can make a significant difference for your business. Could we schedule a brief call to discuss how we can tailor our services to meet your current needs?"
+
+Closing:
+"Thank you for considering CEVauto once again. We're here to support your success and look forward to the opportunity to work with you. Please feel free to reach out at any time. Have a great day!"
+
+'''
     try:
         user_sheet = sheet.worksheet('Users')
         usernames = user_sheet.col_values(2)  # Assuming username is in column B
         is_valid = username in usernames
         
+        # Initialize user's customer tracking if valid
         if is_valid:
-            # Store username in active sessions and initialize page state
-            active_users[username] = datetime.now()
-            page_states[username] = 1  # Start at page 1
-        
+            user_customers[username] = None
+            
         return {
+            verify_output: telescripts if is_valid else "Invalid username",
             customer_section: gr.update(visible=is_valid),
-            update_section: gr.update(visible=is_valid),
-            search_section: gr.update(visible=is_valid),
-            username_display: gr.update(value=f"Logged in as: {username}" if is_valid else "")
+            update_section: gr.update(visible=is_valid)
         }
     except Exception as e:
         return {
+            verify_output: f"Error: {str(e)}",
             customer_section: gr.update(visible=False),
-            update_section: gr.update(visible=False),
-            search_section: gr.update(visible=False),
-            username_display: gr.update(value=f"Error: {str(e)}")
+            update_section: gr.update(visible=False)
         }
 
-def get_best_customer(username):
-    """Get customers with highest RFM scores based on current page"""
-    if username not in active_users:
+def get_all_customers(username):
+    """Get all customers assigned to the username"""
+    if not username:
         return "Please verify your username first"
-
-    try:
-        # Get current page number and increment it
-        current_page = page_states.get(username, 1)
-        page_states[username] = current_page + 1
-
-        customer_sheet = sheet.worksheet('Customers')
-        all_records = customer_sheet.get_all_records()
-
-        # Filter records for current user
-        user_records = [r for r in all_records if r.get('username') == username]
-
-        # Calculate RFM scores for all customers
-        for record in user_records:
-            record['rfm_score'] = calculate_rfm_score(
-                record.get('latest_meeting_date', ''),
-                record.get('visit_times', 0),
-                record.get('call_times', 0),
-                record.get('monetary_value', 1)
-            )
-
-        if not user_records:
-            return "No customers found for your account"
-
-        # Sort by RFM score
-        sorted_customers = sorted(user_records, key=lambda x: x['rfm_score'], reverse=True)
         
-        # Calculate start and end indices for current page
-        start_idx = (current_page - 1) * 10
-        end_idx = start_idx + 10
-        page_customers = sorted_customers[start_idx:end_idx]
-
-        if not page_customers:
-            # If no more customers, reset to page 1
-            page_states[username] = 1
-            page_customers = sorted_customers[:10]
-            current_page = 1
-
-        result = f"Best Customers to Visit (Page {current_page}):\n\n"
-        start_num = (current_page - 1) * 10 + 1
-        for i, customer in enumerate(page_customers, start_num):
-            result += f"""#{i}
-Phone: {customer.get('Phone', 'N/A')}
-Company: {customer.get('Company', 'N/A')}
-Name: {customer.get('Name', 'N/A')}
-RFM Score: {customer.get('rfm_score', 0)}
-Call Times: {customer.get('call_times', 0)}
-Visit Times: {customer.get('visit_times', 0)}
-Latest Meeting: {customer.get('latest_meeting_date', 'N/A')}
-Monetary Value (1-5): {customer.get('monetary_value', 1)}
-Country: {customer.get('Country', 'N/A')}
-Products: {customer.get('Products', 'N/A')}
-Remarks: {customer.get('Remark', 'N/A')}
-------------------------\n"""
-
-        return result
+    try:
+        with sheet_lock:
+            customer_sheet = sheet.worksheet('Customers')
+            all_records = customer_sheet.get_all_records()
+            
+            # Filter records where username matches
+            user_records = [record for record in all_records if record.get('username') == username]
+            
+            if not user_records:
+                return "No customers found for your username"
+                
+            # Format all customer records
+            result = "Your Customers:\n\n"
+            for record in user_records:
+                result += f"""
+Phone: {record['Phone']}
+Company: {record['Company']}
+Name: {record['Name']}
+Status: {record.get('status', '')}
+Country: {record.get('Country', '')}
+Products: {record.get('Products', '')}
+Remark: {record.get('Remark', '')}
+------------------------
+"""
+            return result
+            
     except Exception as e:
         return f"Error: {str(e)}"
 
-def get_all_customers(username):
-    """Get all customers for the current user"""
-    if username not in active_users:
+def get_customer_info(username):
+    """Get and display customer information, update get column"""
+    if not username:
         return "Please verify your username first"
-
+        
     try:
-        customer_sheet = sheet.worksheet('Customers')
-        all_records = customer_sheet.get_all_records()
-
-        # Filter records for current user
-        user_records = [r for r in all_records if r.get('username') == username]
-
-        # Calculate RFM scores for all customers
-        for record in user_records:
-            record['rfm_score'] = calculate_rfm_score(
-                record.get('latest_meeting_date', ''),
-                record.get('visit_times', 0),
-                record.get('call_times', 0),
-                record.get('monetary_value', 1)
-            )
-
-        if not user_records:
-            return "No customers found for your account"
-
-        # Sort by RFM score
-        sorted_customers = sorted(user_records, key=lambda x: x['rfm_score'], reverse=True)
-
-        result = f"All Customers (Total: {len(sorted_customers)}):\n\n"
-        for i, customer in enumerate(sorted_customers, 1):
-            result += f"""#{i}
-Phone: {customer.get('Phone', 'N/A')}
-Company: {customer.get('Company', 'N/A')}
-Name: {customer.get('Name', 'N/A')}
-RFM Score: {customer.get('rfm_score', 0)}
-Call Times: {customer.get('call_times', 0)}
-Visit Times: {customer.get('visit_times', 0)}
-Latest Meeting: {customer.get('latest_meeting_date', 'N/A')}
-Monetary Value (1-5): {customer.get('monetary_value', 1)}
-Country: {customer.get('Country', 'N/A')}
-Products: {customer.get('Products', 'N/A')}
-Remarks: {customer.get('Remark', 'N/A')}
-------------------------\n"""
-
-        return result
+        with sheet_lock:
+            customer_sheet = sheet.worksheet('Customers')
+            all_records = customer_sheet.get_all_records()
+
+            # Find first row where 'get' is empty
+            for idx, record in enumerate(all_records):
+                if not record['get']:
+                    # Update 'get' column to 'yes'
+                    customer_sheet.update_cell(idx + 2, 4, 'yes')  # +2 for header row and 1-based index
+
+                    # Store customer info for this user session
+                    user_customers[username] = {
+                        'row': idx + 2,
+                        'Phone': record['Phone'],
+                        'Company': record['Company'],
+                        'Name': record['Name']
+                    }
+
+                    # Return customer information
+                    return f"""
+                    Phone: {record['Phone']}
+                    Company: {record['Company']}
+                    Name: {record['Name']}
+                    """
+
+            return "No available customer records found"
     except Exception as e:
         return f"Error: {str(e)}"
 
-def search_customer(username, company_name):
-    """Search for a customer by company name"""
-    if username not in active_users:
+def search_customer_by_phone(username, phone):
+    """Search for a customer by phone number and assign to current user"""
+    if not username:
         return "Please verify your username first"
         
-    if not company_name:
-        return "Please enter a company name to search"
+    if not phone:
+        return "Please enter a phone number"
         
     try:
-        customer_sheet = sheet.worksheet('Customers')
-        all_records = customer_sheet.get_all_records()
-        
-        # Find matching companies for current user (case-insensitive)
-        matches = [record for record in all_records 
-                  if company_name in record['Company']]
-        
-        if not matches:
-            return "No matching companies found"
+        with sheet_lock:
+            customer_sheet = sheet.worksheet('Customers')
+            all_records = customer_sheet.get_all_records()
+            
+            # Find customer by phone number
+            for idx, record in enumerate(all_records):
+                if str(record['Phone']) == str(phone):
+                    # Store customer info for this user session
+                    user_customers[username] = {
+                        'row': idx + 2,  # +2 for header row and 1-based index
+                        'Phone': record['Phone'],
+                        'Company': record['Company'],
+                        'Name': record['Name']
+                    }
+                    
+                    return f"""
+                    Phone: {record['Phone']}
+                    Company: {record['Company']}
+                    Name: {record['Name']}
+                    Status: {record.get('status', '')}
+                    Country: {record.get('Country', '')}
+                    Products: {record.get('Products', '')}
+                    Remark: {record.get('Remark', '')}
+                    """
+            
+            return "No customer found with this phone number"
             
-        # Format results
-        result = "Search Results:\n\n"
-        for customer in matches:
-            rfm_score = calculate_rfm_score(
-                customer.get('latest_meeting_date', ''),
-                customer.get('visit_times', 0),
-                customer.get('call_times', 0),
-                customer.get('monetary_value', 1)
-            )
-            result += f"""Phone: {customer.get('Phone', 'N/A')}
-Company: {customer.get('Company', 'N/A')}
-Name: {customer.get('Name', 'N/A')}
-RFM Score: {rfm_score}
-Call Times: {customer.get('call_times', 0)}
-Visit Times: {customer.get('visit_times', 0)}
-Latest Meeting: {customer.get('latest_meeting_date', 'N/A')}
-Monetary Value (1-5): {customer.get('monetary_value', 1)}
-Country: {customer.get('Country', 'N/A')}
-Products: {customer.get('Products', 'N/A')}
-Remarks: {customer.get('Remark', 'N/A')}
-------------------------\n"""
-        
-        return result
     except Exception as e:
         return f"Error: {str(e)}"
 
-def update_customer_info(username, selected_company, call_times, visit_times, 
-                        latest_meeting, monetary_value, country, products, remarks):
-    """Update customer information"""
-    if username not in active_users:
-        return "Please verify your username first"
-        
-    if not selected_company:
-        return "Please select a company first"
-        
+def get_current_customer_info(username):
+    """Get the current customer's information"""
     try:
+        if username not in user_customers or user_customers[username] is None:
+            return "No current customer selected"
+
         customer_sheet = sheet.worksheet('Customers')
-        all_records = customer_sheet.get_all_records()
-        
-        # Find the customer row for current user
-        customer_row = None
-        customer_data = None
-        for idx, record in enumerate(all_records):
-            if (record['Company'].lower() == selected_company.lower() and 
-                record.get('username') == username):
-                customer_row = idx + 2  # +2 for header row and 1-based index
-                customer_data = record
-                break
-        
-        if not customer_row:
-            return "Company not found in your records"
-        
-        # Only update fields that have new values
-        updates = []
-        if call_times:
-            updates.append(('call_times', call_times, 5))
-        if visit_times:
-            updates.append(('visit_times', visit_times, 6))
-        if latest_meeting:
-            updates.append(('latest_meeting_date', latest_meeting, 7))
-        if monetary_value:
-            updates.append(('monetary_value', monetary_value, 8))
-        if country:
-            updates.append(('Country', country, 9))
-        if products:
-            updates.append(('Products', products, 10))
-        if remarks:
-            updates.append(('Remark', remarks, 11))
-        
-        # Calculate new RFM score using new or existing values
-        rfm_score = calculate_rfm_score(
-            latest_meeting or customer_data.get('latest_meeting_date', ''),
-            visit_times or customer_data.get('visit_times', 0),
-            call_times or customer_data.get('call_times', 0),
-            monetary_value or customer_data.get('monetary_value', 1)
-        )
-        updates.append(('rfm_score', rfm_score, 12))
-        
-        # Update only changed fields
-        for field, value, col in updates:
-            customer_sheet.update_cell(customer_row, col, value)
-            
-        return f"Successfully updated customer information. New RFM score: {rfm_score}"
+        row_values = customer_sheet.row_values(user_customers[username]['row'])
+
+        return f"""
+        Phone: {row_values[0]}
+        Company: {row_values[1]}
+        Name: {row_values[2]}
+        Status: {row_values[5] if len(row_values) > 5 else ''}
+        Country: {row_values[6] if len(row_values) > 6 else ''}
+        Products: {row_values[7] if len(row_values) > 7 else ''}
+        Remark: {row_values[8] if len(row_values) > 8 else ''}
+        """
     except Exception as e:
         return f"Error: {str(e)}"
 
+def submit_customer_update(username, status, country, products, remark):
+    """Submit customer information updates and refresh display"""
+    if username not in user_customers or user_customers[username] is None:
+        return "No customer selected", "Please get a customer first"
+
+    try:
+        with sheet_lock:
+            customer_sheet = sheet.worksheet('Customers')
+
+            # Update the fields in the correct row
+            updates = [
+                ('username', username, 5),  # Column E
+                ('status', status, 6),  # Column F
+                ('Country', country, 7),  # Column G
+                ('Products', products, 8),  # Column H
+                ('Remark', remark, 9)  # Column I
+            ]
+
+            for field, value, col in updates:
+                customer_sheet.update_cell(user_customers[username]['row'], col, value)
+
+            # Get updated customer information
+            updated_info = get_current_customer_info(username)
+            return "Update successful", updated_info
+    except Exception as e:
+        return f"Error: {str(e)}", "Error occurred while updating"
+
 # Create Gradio interface
 with gr.Blocks() as app:
     gr.Markdown(
         """
-        # Customer Management System
-        ### Efficiently manage and track customer information
+        # CEVauto Customer Management System
+        ### Efficiently manage and update customer information
         ---
         """
     )
@@ -320,120 +264,89 @@ with gr.Blocks() as app:
     # User Verification Section
     gr.Markdown("### 1. User Verification")
     with gr.Group():
-        username_input = gr.Textbox(label="Username", placeholder="Enter your username")
-        verify_button = gr.Button("Verify", variant="primary")
-        username_display = gr.Textbox(label="Status", interactive=False)
+        with gr.Row():
+            with gr.Column(scale=3):
+                username_input = gr.Textbox(label="Username", placeholder="Enter your username")
+            with gr.Column(scale=1):
+                verify_button = gr.Button("Verify", variant="primary")
+        verify_output = gr.Textbox(label="Cold Call Scripts", interactive=False)
+
+    gr.Markdown("---")
 
     # Customer Information Section
     with gr.Group(visible=False) as customer_section:
-        gr.Markdown("### 2. Best Customer Information")
+        gr.Markdown("### 2. Customer Information")
+        with gr.Row():
+            with gr.Column(scale=2):
+                phone_search = gr.Textbox(label="Search by Phone", placeholder="Enter phone number")
+            with gr.Column(scale=1):
+                search_button = gr.Button("Search Customer", variant="secondary")
         with gr.Row():
-            get_button = gr.Button("Get Best Customer (Next Page)", variant="primary")
-            show_all_button = gr.Button("Show All Customers", variant="secondary")
+            get_button = gr.Button("Get Next Customer", variant="primary")
+            show_all_button = gr.Button("Show All My Customers", variant="secondary")
         customer_info = gr.Textbox(
             label="Customer Details",
             interactive=False,
-            lines=12
+            lines=15
         )
 
-    # Search Section
-    with gr.Group(visible=False) as search_section:
-        gr.Markdown("### 3. Search Customer")
-        search_input = gr.Textbox(label="Company Name", placeholder="Enter company name to search")
-        search_button = gr.Button("Search", variant="primary")
-        search_results = gr.Textbox(
-            label="Search Results",
-            interactive=False,
-            lines=12
-        )
+        gr.Markdown("---")
 
-    # Update Section
+    # Update Status Section
     with gr.Group(visible=False) as update_section:
-        gr.Markdown("### 4. Update Customer Information")
-        selected_company = gr.Dropdown(
-            label="Select Company",
-            choices=[],  # Will be populated after search
-            interactive=True
-        )
-        call_times_input = gr.Number(label="Call Times")
-        visit_times_input = gr.Number(label="Visit Times")
-        latest_meeting_input = gr.Textbox(
-            label="Latest Meeting Date",
-            placeholder="YYYY-MM-DD (e.g., 2024-01-31)"
-        )
-        monetary_input = gr.Slider(
-            label="Monetary Value (1-5)",
-            minimum=1,
-            maximum=5,
-            step=1
-        )
-        country_input = gr.Textbox(label="Country")
-        products_input = gr.Textbox(label="Products")
-        remarks_input = gr.Textbox(label="Remarks", lines=3)
-        
-        update_button = gr.Button("Update Customer", variant="primary")
-        update_status = gr.Textbox(label="Update Status", interactive=False)
+        gr.Markdown("### 3. Update Customer Status")
+        with gr.Row():
+            with gr.Column():
+                status_input = gr.Dropdown(
+                    choices=["Unconnected", "Invalid", "Not interested", "interested", "Willing"],
+                    label="Status",
+                    value="Unconnected"
+                )
+                country_input = gr.Textbox(label="Country", placeholder="Enter customer's country")
+            with gr.Column():
+                products_input = gr.Textbox(label="Products", placeholder="Enter relevant products")
+                remark_input = gr.Textbox(label="Remark", placeholder="Add any additional notes", lines=2)
 
-    def update_company_choices(username, search_text):
-        if not username or not search_text:
-            return gr.update(choices=[])
-        try:
-            customer_sheet = sheet.worksheet('Customers')
-            all_records = customer_sheet.get_all_records()
-            matches = [r['Company'] for r in all_records 
-                      if search_text.lower() in r['Company'].lower()
-                      and r.get('username') == username]
-            return gr.update(choices=matches)
-        except Exception:
-            return gr.update(choices=[])
+        with gr.Row():
+            submit_button = gr.Button("Submit Update", variant="primary")
+            submit_output = gr.Textbox(label="Update Status", interactive=False)
 
     # Event handlers
     verify_button.click(
         fn=verify_username,
         inputs=username_input,
-        outputs=[customer_section, update_section, search_section, username_display]
+        outputs=[verify_output, customer_section, update_section]
     )
 
     get_button.click(
-        fn=get_best_customer,
+        fn=get_customer_info,
         inputs=username_input,
         outputs=customer_info
     )
-
+    
     show_all_button.click(
         fn=get_all_customers,
         inputs=username_input,
         outputs=customer_info
     )
-
+    
     search_button.click(
-        fn=search_customer,
-        inputs=[username_input, search_input],
-        outputs=search_results
-    )
-
-    # Update company dropdown when searching
-    search_input.change(
-        fn=update_company_choices,
-        inputs=[username_input, search_input],
-        outputs=selected_company
+        fn=search_customer_by_phone,
+        inputs=[username_input, phone_search],
+        outputs=customer_info
     )
 
-    update_button.click(
-        fn=update_customer_info,
+    submit_button.click(
+        fn=submit_customer_update,
         inputs=[
             username_input,
-            selected_company,
-            call_times_input,
-            visit_times_input,
-            latest_meeting_input,
-            monetary_input,
+            status_input,
             country_input,
             products_input,
-            remarks_input
+            remark_input
         ],
-        outputs=update_status
+        outputs=[submit_output, customer_info]
     )
 
-# Launch the app
+# Launch the app with authentication
 app.launch()