Update app.py

app.py

@@ -2,24 +2,6 @@ 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"
@@ -114,41 +96,6 @@ 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
 
@@ -188,4 +135,4 @@ def create_interface():
 
 if __name__ == "__main__":
     demo = create_interface()
-    demo.launch()
+    demo.launch()