Update app.py

app.py

@@ -576,7 +576,7 @@ async def generate_campaign_sequences(campaign_id: int, request: GenerateSequenc
     return {
         "ok": True,
         "message": "Sequences generated and saved successfully",
-        "generated_sequences": [seq.dict() for seq in generated_sequences],
+        "generated_sequences": [seq for seq in generated_sequences],
         "save_result": result
     }
 
@@ -836,6 +836,16 @@ def generate_template_welcome_closing_messages(lead_data: Dict[str, Any]) -> Dic
     }
 
 async def generate_sequences_with_llm(job_description: str) -> List[CampaignSequence]:
+    class email_seq(BaseModel):
+        subject: str = Field(description="Subject line for the email")
+        body: str = Field(description="Body of the email")
+    class structure(BaseModel):
+        introduction: email_seq = Field(description="Email sequence for sequence 1 asking for consent and interest in the role")
+        email_sequence_2: email_seq = Field(description="Email sequence for sequence 2 following up on updates and next steps")
+        email_sequence_3: email_seq = Field(description="Email sequence for sequence 3 Another variant on following up on updates and next steps")
+
+
+
     """Generate email sequences using LangChain and OpenAI based on job description"""
     
     if not LANGCHAIN_AVAILABLE:
@@ -848,46 +858,34 @@ async def generate_sequences_with_llm(job_description: str) -> List[CampaignSequ
             return await generate_template_sequences(job_description)
         
         llm = ChatOpenAI(
-            model="gpt-4",
+            model="gpt-4o",
             temperature=0.7,
             openai_api_key=openai_api_key
         )
+        str_llm = llm.with_structured_output(structure)
         
-        system_prompt = """You are an expert email sequence generator for recruitment campaigns. 
-        
+        system_prompt = """You are an expert email sequence template generator for recruitment campaigns.
+
         Generate ONLY the subject lines and email body content for 3 professional email sequences.
-        
+        Write the email on behalf of Ali Taghikhani, CEO SRN
+        In the templates use placeholders like, welcome_message, closing message, first_name, company,title only.
+
         Email Sequence Structure:
-        1. INTRODUCTION (Day 1): Ask for consent and interest in the role
-        2. OUTREACH (Day 3): Provide detailed job information  
+        1. INTRODUCTION (Day 1): Ask for consent and interest in the role, In the starting use the welcome message placeholder, and in the end use closing message template along with the name and title of sender.placeholders must be used like this {{first_name}}
+        2. OUTREACH (Day 3): Provide detailed job information
         3. FOLLOW-UP (Day 5): Follow up on updates and next steps
-        
+
         Requirements:
-        - First sequence needs 2 A/B testing variants (A and B)
-        - Second and third sequences are follow-ups (no subject line needed)
+        - First sequence will only ask about the consent and interest in the role, no other information is needed.
+        - Second and third sequences are follow-ups (no subject line needed)  in the 3rd sequence try providing some information about the role and the company.
         - All emails should be HTML formatted with proper <br> tags
         - Professional but friendly tone
         - Include clear call-to-actions
         - Focus on building consent and trust
-        
+
         Respond with ONLY a JSON object containing the email content:
-        {
-            "sequence1_variant_a": {
-                "subject": "Subject line for variant A",
-                "body": "HTML formatted email body for variant A"
-            },
-            "sequence1_variant_b": {
-                "subject": "Subject line for variant B", 
-                "body": "HTML formatted email body for variant B"
-            },
-            "sequence2": {
-                "body": "HTML formatted email body for outreach"
-            },
-            "sequence3": {
-                "body": "HTML formatted email body for follow-up"
-            }
-        }
-        
+
+
         IMPORTANT: Respond with ONLY valid JSON. No additional text."""
         
         prompt_template = ChatPromptTemplate.from_messages([
@@ -896,19 +894,121 @@ async def generate_sequences_with_llm(job_description: str) -> List[CampaignSequ
         ])
         
         messages = prompt_template.format_messages(job_description=job_description)
-        response = await llm.ainvoke(messages)
+        response = await str_llm.ainvoke(messages)
         
         try:
-            content = response.content.strip()
+            # Handle the response structure correctly
+            sequences = []
             
-            if content.startswith("```json"):
-                content = content[7:]
-            if content.endswith("```"):
-                content = content[:-3]
+            # Sequence 1: Introduction with A/B variants
+            if hasattr(response, 'introduction') and response.introduction:
+                # Check if body is a string or dict
+                intro_body = response.introduction.body
+                if isinstance(intro_body, dict):
+                    # If it's a dict, extract the content
+                    intro_body = str(intro_body)
+                
+                sequences.append(CampaignSequence(
+                    seq_number=1,
+                    seq_delay_details=SeqDelayDetails(delay_in_days=1),
+                    seq_variants=[
+                        SeqVariant(
+                            subject=response.introduction.subject,
+                            email_body=intro_body,
+                            variant_label="A"
+                        )
+                    ]
+                ))
+            
+            # Sequence 2: Outreach
+            if hasattr(response, 'email_sequence_2') and response.email_sequence_2:
+                seq2_body = response.email_sequence_2.body
+                if isinstance(seq2_body, dict):
+                    seq2_body = str(seq2_body)
+                
+                sequences.append(CampaignSequence(
+                    seq_number=2,
+                    seq_delay_details=SeqDelayDetails(delay_in_days=3),
+                    subject="",
+                    email_body=seq2_body
+                ))
+            
+            # Sequence 3: Follow-up
+            if hasattr(response, 'email_sequence_3') and response.email_sequence_3:
+                seq3_body = response.email_sequence_3.body
+                if isinstance(seq3_body, dict):
+                    seq3_body = str(seq3_body)
+                
+                sequences.append(CampaignSequence(
+                    seq_number=3,
+                    seq_delay_details=SeqDelayDetails(delay_in_days=5),
+                    subject="",
+                    email_body=seq3_body
+                ))
+            
+            # Fill with templates if needed
+            while len(sequences) < 3:
+                if len(sequences) == 0:
+                    sequences.append(CampaignSequence(
+                        seq_number=1,
+                        seq_delay_details=SeqDelayDetails(delay_in_days=1),
+                        seq_variants=[
+                            SeqVariant(
+                                subject=f"Quick question about {job_description}",
+                                email_body=f"""<p>Hi there,<br><br>
+I came across your profile and noticed your experience in {job_description}. 
+I'm reaching out because we have some exciting opportunities that might be a great fit for your background.<br><br>
+Before I share more details, I wanted to ask: Are you currently open to exploring new opportunities in this space?<br><br>
+Would you be interested in hearing more about the roles we have available?<br><br>
+Best regards,<br>
+[Your Name]</p>""",
+                                variant_label="A"
+                            ),
+                            SeqVariant(
+                                subject=f"Interested in {job_description} opportunities?",
+                                email_body=f"""<p>Hello,<br><br>
+I hope this message finds you well. I'm a recruiter specializing in {job_description} positions.<br><br>
+I'd love to connect and share some opportunities that align with your expertise. 
+Are you currently open to exploring new roles in this space?<br><br>
+If so, I can send you specific details about the positions we have available.<br><br>
+Thanks,<br>
+[Your Name]</p>""",
+                                variant_label="B"
+                            )
+                        ]
+                    ))
+                elif len(sequences) == 1:
+                    sequences.append(CampaignSequence(
+                        seq_number=2,
+                        seq_delay_details=SeqDelayDetails(delay_in_days=3),
+                        subject="",
+                        email_body=f"""<p>Hi,<br><br>
+Thanks for your interest! Here are more details about the {job_description} opportunities:<br><br>
+<strong>Role Details:</strong><br>
+• [Specific responsibilities]<br>
+• [Required skills and experience]<br>
+• [Team and company information]<br><br>
+<strong>Benefits:</strong><br>
+• [Compensation and benefits]<br>
+• [Growth opportunities]<br>
+• [Work environment]<br><br>
+Would you be interested in a quick call to discuss this role in more detail?<br><br>
+Best regards,<br>
+[Your Name]</p>"""
+                    ))
+                elif len(sequences) == 2:
+                    sequences.append(CampaignSequence(
+                        seq_number=3,
+                        seq_delay_details=SeqDelayDetails(delay_in_days=5),
+                        subject="",
+                        email_body=f"""<p>Hi,<br><br>
+Just wanted to follow up on the {job_description} opportunity I shared.<br><br>
+Have you had a chance to review the information? I'd love to hear your thoughts and answer any questions.<br><br>
+If you're interested, I can help schedule next steps. If not, no worries at all!<br><br>
+Thanks for your time!<br>
+[Your Name]</p>"""
+                    ))
             
-            content = content.strip()
-            parsed_data = json.loads(content)
-            sequences = create_sequences_from_content(parsed_data, job_description)
             return sequences
             
         except Exception as parse_error: