Update app.py

app.py

@@ -1,4 +1,4 @@
-import gradio as gr
+from flask import Flask, render_template, request, redirect, url_for
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from simple_salesforce import Salesforce
@@ -6,6 +6,8 @@ import os
 from datetime import datetime
 import logging
 
+app = Flask(__name__)
+
 # Configure logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
@@ -177,7 +179,7 @@ def fetch_and_process_salesforce_data():
             role = supervisor.get('Role__c', 'Supervisor')
 
             # Fetch other fields
-            project_id = project.get('Name', '')  # Use Name instead of Project_ID__c
+            project_id = project.get('Name', '')
             milestones = project.get('Milestones__c', 'No milestones')
 
             # Fetch Reflection from Supervisor_AI_Coaching__c
@@ -202,7 +204,7 @@ def fetch_and_process_salesforce_data():
                 'Supervisor_ID__c': project.get('Supervisor__c', ''),
                 'Daily_Checklist__c': checklist,
                 'Suggested_Tips__c': suggestions,
-                'Motivational_Quote__c': quote,  # Assumed field for Quote
+                'Motivational_Quote__c': quote,
                 'Reflection_Log__c': reflection,
                 'Last_Refresh_Date__c': datetime.utcnow().isoformat(),
                 'Engagement_Score__c': 50 if checklist != 'Not generated.' else 0,
@@ -221,52 +223,54 @@ def fetch_and_process_salesforce_data():
         logger.error(f"Error processing Salesforce data: {e}")
         return f"Error processing Salesforce data: {e}"
 
-def create_interface():
-    """Create Gradio interface for manual testing."""
-    with gr.Blocks() as demo:
-        gr.Markdown("### Construction Supervisor AI Coach")
-        with gr.Row():
-            role = gr.Dropdown(choices=["Supervisor", "Foreman", "Project Manager"], label="Role", value="Supervisor")
-            project_id = gr.Textbox(label="Project ID", placeholder="e.g., PROJ-123")
-        milestones = gr.Textbox(
-            label="Milestones (comma-separated)",
-            placeholder="e.g., Foundation complete, Framing started, Roof installed"
-        )
-        reflection = gr.Textbox(
-            label="Reflection",
-            lines=3,
-            placeholder="e.g., Facing delays due to weather and equipment issues."
-        )
-        with gr.Row():
-            submit = gr.Button("Generate")
-            clear = gr.Button("Clear")
-            sf_button = gr.Button("Process Salesforce Data")
-        checklist_output = gr.Textbox(label="Checklist", lines=4)
-        suggestions_output = gr.Textbox(label="Suggestions", lines=4)
-        quote_output = gr.Textbox(label="Quote", lines=2)
-        sf_output = gr.Textbox(label="Salesforce Processing Result", lines=2)
-        
-        submit.click(
-            fn=generate_outputs,
-            inputs=[role, project_id, milestones, reflection],
-            outputs=[checklist_output, suggestions_output, quote_output]
-        )
-        clear.click(
-            fn=lambda: ("Supervisor", "", "", ""),
-            inputs=None,
-            outputs=[role, project_id, milestones, reflection]
-        )
-        sf_button.click(
-            fn=fetch_and_process_salesforce_data,
-            inputs=None,
-            outputs=sf_output
-        )
-    
-    return demo
+@app.route('/', methods=['GET', 'POST'])
+def index():
+    # Default values
+    role = "Supervisor"
+    project_id = ""
+    milestones = ""
+    reflection = ""
+    checklist = ""
+    suggestions = ""
+    quote = ""
+    sf_result = ""
 
-if __name__ == "__main__":
-    try:
-        demo = create_interface()
-        demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
-    except Exception as e:
-        logger.error(f"Error launching Gradio interface: {e}")
+    if request.method == 'POST':
+        action = request.form.get('action')
+
+        if action == 'generate':
+            # Get form inputs
+            role = request.form.get('role', 'Supervisor')
+            project_id = request.form.get('project_id', '')
+            milestones = request.form.get('milestones', '')
+            reflection = request.form.get('reflection', '')
+
+            # Generate outputs
+            checklist, suggestions, quote = generate_outputs(role, project_id, milestones, reflection)
+
+        elif action == 'clear':
+            # Clear the form
+            role = "Supervisor"
+            project_id = ""
+            milestones = ""
+            reflection = ""
+            checklist = ""
+            suggestions = ""
+            quote = ""
+
+        elif action == 'process_salesforce':
+            # Process Salesforce data
+            sf_result = fetch_and_process_salesforce_data()
+
+    return render_template('index.html', 
+                           role=role, 
+                           project_id=project_id, 
+                           milestones=milestones, 
+                           reflection=reflection, 
+                           checklist=checklist, 
+                           suggestions=suggestions, 
+                           quote=quote, 
+                           sf_result=sf_result)
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=5000, debug=True)