Update app.py

app.py

@@ -48,31 +48,51 @@ def generate_text(prompt):
         logger.error(f"Error in generate_text: {type(e).__name__}: {str(e)}")
         return f"An error occurred: {type(e).__name__}: {str(e)}"
 
+try:
+    generator = pipeline('text-generation', model=model_name)
+    logger.info(f"Successfully loaded model: {model_name}")
+except Exception as e:
+    logger.error(f"Failed to load model: {type(e).__name__}: {str(e)}")
+    raise
+
 def generate_email(industry, recipient_role, company_details):
     try:
-        logger.info(f"Generating email for {industry}, {recipient_role}")
-        prompt = f"""Write a professional cold outreach email using the following details:
+        prompt = f"""Generate a professional cold outreach email using the following details:
         Industry: {industry}
         Recipient Role: {recipient_role}
         Company Details: {company_details}
 
-        The email should:
-        1. Have a clear and concise subject line
-        2. Start with a personalized greeting
-        3. Briefly introduce yourself and your company
-        4. Explain the value proposition relevant to the recipient's industry and role
-        5. Include a clear call to action
-        6. End with a professional closing
+        Format the email as follows:
+        Subject: [Insert catchy subject line related to the industry and recipient role]
+
+        Dear [Recipient's Name],
+
+        [Introduction paragraph: Briefly introduce yourself and your company]
+
+        [Value proposition paragraph: Explain how your company can benefit the recipient, citing specific details from the company information]
+
+        [Call to action paragraph: Suggest a meeting or call to discuss further]
+
+        [Closing paragraph: Thank the recipient and provide your contact information]
+
+        Best regards,
+        [Your Name]
+        [Your Title]
+        [Your Company]
 
         Email:
         """
-        response = generate_text(prompt)
-        if isinstance(response, str) and response.startswith("An error occurred:"):
-            logger.error(f"Error response from generate_text: {response}")
-            return "I apologize, but I'm currently unable to generate an email. There might be an issue with the AI model access. Please try again later or contact support."
-        return response
+        
+        response = generator(prompt, max_length=800, num_return_sequences=1, temperature=0.7)
+        generated_text = response[0]['generated_text']
+        
+        # Remove the prompt from the generated text
+        email_content = generated_text[generated_text.find("Subject:"):].strip()
+        
+        logger.info(f"Generated email for {industry}, {recipient_role}")
+        return email_content
     except Exception as e:
-        logger.error(f"Error in generate_email: {str(e)}")
+        logger.error(f"Error in generate_email: {type(e).__name__}: {str(e)}")
         return "I apologize, but an unexpected error occurred. Please try again later or contact support."
 
 def test_model_connection():