Update app.py

app.py

@@ -7,14 +7,15 @@ import re
 import random
 from openai import OpenAI
 
-# Initialize OpenAI client securely (use Hugging Face secret)
+# Initialize OpenAI client securely
 client = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
 
 # Convert a CRM row into readable text for embeddings
 def row_to_text(row):
     return (
         f"Clinic: {row.get('Account_Name', 'N/A')}\n"
-        f"Industry: {row.get('Industry', 'Dental')}\n"
+        f"Interaction Focus: {row.get('Recent_Interaction_Focus', 'N/A')}\n"
+        f"Interaction Notes: {row.get('Recent_Interaction_Notes', 'N/A')}\n"
         f"Region: {row.get('Region', 'Unknown')}\n"
         f"Account Manager: {row.get('Account_Manager', 'Unknown')}\n"
         f"Revenue: €{row.get('Revenue', 'N/A')}\n"
@@ -45,21 +46,29 @@ def contextual_pitch_assistant(csv_file, query, sender_name):
 
     q_emb = np.array([get_embedding(query)]).astype("float32")
     D, I = index.search(q_emb, 3)
-    retrieved = [text_chunks[i] for i in I[0]]
-
-    # Extract name and clinic safely first
-    try:
-        account_manager = retrieved[0].split("Account Manager: ")[1].split("\n")[0].strip()
-    except Exception:
-        account_manager = "team"
-    try:
-        account_name = retrieved[0].split("Clinic: ")[1].split("\n")[0].strip()
-    except Exception:
-        account_name = "your clinic"
+    retrieved = [df.iloc[i] for i in I[0]]
+
+    # Extract contextual fields
+    row = retrieved[0]
+    account_manager = row.get("Account_Manager", "team")
+    account_name = row.get("Account_Name", "your clinic")
+    focus = str(row.get("Recent_Interaction_Focus", "")).lower()
+    notes = str(row.get("Recent_Interaction_Notes", ""))
 
     if not sender_name.strip():
         sender_name = "The Sales Team"
 
+    # Map focus → style instruction
+    if "business" in focus:
+        style_instruction = "Emphasize ROI, efficiency, and competitive advantage."
+    elif "patient" in focus:
+        style_instruction = "Emphasize patient comfort, satisfaction, and outcomes."
+    elif "clinical" in focus:
+        style_instruction = "Emphasize precision, innovation, and clinical quality."
+    else:
+        style_instruction = "Balance clinical, patient, and business value."
+
+    # Build prompt
     prompt = f"""
 You are an expert in B2B email sales and marketing for dental technology solutions.
 
@@ -68,6 +77,8 @@ Write a short, natural-sounding HTML sales email for a dental clinic.
 Rules:
 - Always begin with "Dear {account_manager}," as the greeting.
 - Refer naturally to the clinic "{account_name}" in the body.
+- Adapt the pitch style: {style_instruction}
+- Use this recent interaction note for inspiration: "{notes}"
 - Focus on value, benefits, and a helpful tone — avoid technical jargon or metrics like churn or satisfaction scores.
 - Never include placeholders such as "Your Company Name" or "[Your Name]".
 - End the email with "Best regards," followed by "{sender_name}".
@@ -79,7 +90,7 @@ Query:
 {query}
 
 CRM context (for your understanding, do not copy verbatim):
-{'---'.join(retrieved)}
+{row_to_text(row)}
 """
 
     response = client.chat.completions.create(
@@ -113,7 +124,7 @@ CRM context (for your understanding, do not copy verbatim):
     """
     return html
 
-# Build the Gradio app UI
+# Build Gradio app (unchanged, apart from column updates)
 with gr.Blocks(
     title="Contextual Pitch Assistant for Dental Sales",
     css="""
@@ -152,7 +163,7 @@ with gr.Blocks(
     )
 
     with gr.Row():
-        csv_file = gr.File(label="📂 Upload CRM CSV (5–100 rows)", file_types=[".csv"])
+        csv_file = gr.File(label="📂 Upload CRM CSV (with Recent_Interaction_Focus/Notes)", file_types=[".csv"])
         sender_name = gr.Textbox(
             label="✍️ Who signs the email?", 
             placeholder="e.g. Nuno Camacho, Sales Director", 
@@ -178,4 +189,3 @@ with gr.Blocks(
     )
 
 app.launch()
-