Update app.py

app.py

@@ -2,6 +2,24 @@ import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import datetime
+from simple_salesforce import Salesforce
+import os
+import uuid
+import tempfile
+
+# Salesforce configuration
+SF_USERNAME = os.getenv('SF_USERNAME', 'your_salesforce_username')
+SF_PASSWORD = os.getenv('SF_PASSWORD', 'your_salesforce_password')
+SF_SECURITY_TOKEN = os.getenv('SF_SECURITY_TOKEN', 'your_salesforce_security_token')
+SF_DOMAIN = 'login'  # Use 'test' for sandbox or 'login' for production
+
+# Initialize Salesforce connection
+sf = Salesforce(
+    username=SF_USERNAME,
+    password=SF_PASSWORD,
+    security_token=SF_SECURITY_TOKEN,
+    domain=SF_DOMAIN
+)
 
 # Initialize model and tokenizer (preloading them for quicker response)
 model_name = "distilgpt2"
@@ -96,6 +114,41 @@ def generate_outputs(role, project_id, milestones, reflection):
         quote_start = generated_text.find("Quote:") + len("Quote:")
         quote = generated_text[quote_start:].strip()
 
+    # Generate a file with the processed output
+    output_content = f"""Checklist:
+{checklist}
+
+Suggestions:
+{suggestions}
+
+Quote:
+{quote}
+"""
+    # Create a temporary file
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.txt', mode='w', encoding='utf-8')
+    temp_file.write(output_content)
+    temp_file.close()
+    
+    # Simulate a download URL (in production, upload to a file hosting service like Salesforce Content or AWS S3)
+    file_name = f"supervisor_coaching_{uuid.uuid4()}.txt"
+    download_url = f"/tmp/{file_name}"  # Placeholder URL; replace with actual file hosting URL in production
+    os.rename(temp_file.name, os.path.join(tempfile.gettempdir(), file_name))
+
+    # Save to Salesforce Supervisor_AI_Coaching__c object
+    try:
+        sf.Supervisor_AI_Coaching__c.create({
+            'Role__c': role,
+            'Project_ID__c': project_id,
+            'Milestones__c': milestones,
+            'Reflection__c': reflection,
+            'Checklist__c': checklist,
+            'Suggestions__c': suggestions,
+            'Quote__c': quote,
+            'Download_Link__c': download_url
+        })
+    except Exception as e:
+        print(f"Error saving to Salesforce: {str(e)}")
+
     # Return structured outputs
     return checklist, suggestions, quote
 
@@ -135,4 +188,4 @@ def create_interface():
 
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch()
+    demo.launch()