Update app.py

app.py

@@ -9,17 +9,17 @@ from dotenv import load_dotenv
 from fpdf import FPDF
 import shutil
 import html
+import matplotlib.pyplot as plt
+import io
 
 # Load environment variables
 load_dotenv()
 
-# Required env vars check
 required_env_vars = ['SF_USERNAME', 'SF_PASSWORD', 'SF_SECURITY_TOKEN']
 missing_vars = [var for var in required_env_vars if not os.getenv(var)]
 if missing_vars:
     raise EnvironmentError(f"Missing required environment variables: {missing_vars}")
 
-# Load model and tokenizer
 model_name = "distilgpt2"
 tokenizer = AutoTokenizer.from_pretrained(model_name, use_fast=True)
 model = AutoModelForCausalLM.from_pretrained(model_name, low_cpu_mem_usage=True)
@@ -196,6 +196,18 @@ def get_dashboard_data_from_salesforce(supervisor_name, project_id):
     except Exception as e:
         return f"Error loading dashboard: {e}", "", "", ""
 
+def render_bar_chart(completed, total):
+    fig, ax = plt.subplots(figsize=(4, 3))
+    ax.bar(["Completed", "Remaining"], [completed, max(0, total - completed)], color=["green", "gray"])
+    ax.set_title("Checklist Progress")
+    buf = io.BytesIO()
+    plt.tight_layout()
+    fig.savefig(buf, format="png")
+    plt.close(fig)
+    buf.seek(0)
+    encoded = base64.b64encode(buf.read()).decode("utf-8")
+    return f"<img src='data:image/png;base64,{encoded}'/>"
+
 def generate_outputs(role, supervisor_name, project_id, milestones, reflection):
     if not all([role, supervisor_name, project_id, milestones, reflection]):
         return "❗ Please fill all fields.", "", None, ""
@@ -287,6 +299,9 @@ def create_interface():
             def show_dashboard_html(sup_name, proj_id):
                 checklist, suggestions, link, date = get_dashboard_data_from_salesforce(sup_name, proj_id)
                 link_html = f'<a href="{link}" target="_blank">Download PDF</a>' if link else "No report available"
+                completed_tasks = checklist.count("- ")
+                total_tasks = 5
+                chart_html = render_bar_chart(completed_tasks, total_tasks)
                 html_card = f"""
                 <div style='border:1px solid #ccc; border-radius:10px; padding:20px; background-color:#f9f9f9'>
                     <h3>📋 <u>Dashboard Summary</u></h3>
@@ -299,6 +314,9 @@ def create_interface():
                     <h4>💡 <u>Focus Suggestions</u></h4>
                     <pre style='background:#ffe; padding:10px; border-radius:5px;'>{suggestions}</pre>
                     <p>{link_html}</p>
+                    <hr>
+                    <h4>📊 <u>Progress Chart</u></h4>
+                    {chart_html}
                 </div>
                 """
                 return html_card