Update app.py

app.py

@@ -1,91 +1,107 @@
-from flask import Flask, render_template, send_from_directory
-from celery import Celery
-import pandas as pd
-from simple_salesforce import Salesforce
-import os
-
-# Salesforce credentials
-SALESFORCE_USERNAME = "venkatramana@sandbox.com"
-SALESFORCE_PASSWORD = "Seta12345@"
-SALESFORCE_SECURITY_TOKEN = "ZOnDC5se0y3CACNjNSsBWQPl"
-
-# Initialize Flask app
-app = Flask(__name__)
-
-# Initialize Celery
-celery = Celery(app.name, broker='redis://localhost:6379/0')
-
-# Salesforce connection
-def connect_to_salesforce():
-    try:
-        sf = Salesforce(username=SALESFORCE_USERNAME, password=SALESFORCE_PASSWORD, security_token=SALESFORCE_SECURITY_TOKEN)
-        return sf
-    except Exception as e:
-        print(f"Error connecting to Salesforce: {e}")
-        return None
-
-# Fetch quote line items from Salesforce
-def fetch_quote_line_items(sf):
-    try:
-        query = """
-        SELECT Id, Product__c, Quantity__c, Price_c__c
-        FROM Quote_Line_Item__c
-        LIMIT 50
-        """
-        line_items = sf.query(query)
-        return line_items['records']
-    except Exception as e:
-        print(f"Error fetching quote line items: {e}")
-        return None
-
-# Fetch products and calculate totals
-def fetch_products():
-    sf = connect_to_salesforce()
-    if not sf:
-        return []
-
-    line_items = fetch_quote_line_items(sf)
-    products = []
-    for item in line_items:
-        product = item.get('Product__c', 'N/A')
-        quantity = item.get('Quantity__c', 'N/A')
-        price = item.get('Price_c__c', 'N/A')
-        total = float(quantity) * float(price) if quantity and price else 'N/A'
-        products.append({
-            'Product': product,
-            'Quantity': quantity,
-            'Price': price,
-            'Total': total
-        })
-    return products
-
-# Save products to Excel
-def save_to_excel(products, file_path):
-    df = pd.DataFrame(products)
-    df.to_excel(file_path, index=False)
-
-# Celery task for Excel generation
-@celery.task
-def generate_excel_task():
-    products = fetch_products()
-    file_path = "products_data.xlsx"
-    save_to_excel(products, file_path)
-    print(f"Excel file generated at {file_path}")
-
-# Flask routes
-@app.route('/')
-def index():
-    files = [f for f in os.listdir('.') if f.endswith('.xlsx')]
-    return render_template('index.html', files=files)
-
-@app.route('/download/<filename>')
-def download_file(filename):
-    return send_from_directory('.', filename)
-
-@app.route('/generate_excel')
-def generate_excel():
-    generate_excel_task.delay()  # Trigger Celery task
-    return "Excel generation started!"
-
-if __name__ == '__main__':
-    app.run(debug=True)
+from fastapi import FastAPI
+from fastapi.responses import FileResponse
+from fastapi.templating import Jinja2Templates
+from fastapi.staticfiles import StaticFiles
+from starlette.requests import Request
+from celery import Celery
+import pandas as pd
+from simple_salesforce import Salesforce
+import os
+
+# Salesforce credentials
+SALESFORCE_USERNAME = "venkatramana@sandbox.com"
+SALESFORCE_PASSWORD = "Seta12345@"
+SALESFORCE_SECURITY_TOKEN = "ZOnDC5se0y3CACNjNSsBWQPl"
+
+# Initialize FastAPI app
+app = FastAPI()
+
+# Initialize Celery
+celery = Celery("app", broker="redis://localhost:6379/0")
+
+# Set up Jinja2 templates
+templates = Jinja2Templates(directory="templates")
+
+# Serve static files (if needed)
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
+# Salesforce connection
+def connect_to_salesforce():
+    try:
+        sf = Salesforce(
+            username=SALESFORCE_USERNAME,
+            password=SALESFORCE_PASSWORD,
+            security_token=SALESFORCE_SECURITY_TOKEN,
+        )
+        return sf
+    except Exception as e:
+        print(f"Error connecting to Salesforce: {e}")
+        return None
+
+# Fetch quote line items from Salesforce
+def fetch_quote_line_items(sf):
+    try:
+        query = """
+        SELECT Id, Product__c, Quantity__c, Price_c__c
+        FROM Quote_Line_Item__c
+        LIMIT 50
+        """
+        line_items = sf.query(query)
+        return line_items["records"]
+    except Exception as e:
+        print(f"Error fetching quote line items: {e}")
+        return None
+
+# Fetch products and calculate totals
+def fetch_products():
+    sf = connect_to_salesforce()
+    if not sf:
+        return []
+
+    line_items = fetch_quote_line_items(sf)
+    products = []
+    for item in line_items:
+        product = item.get("Product__c", "N/A")
+        quantity = item.get("Quantity__c", "N/A")
+        price = item.get("Price_c__c", "N/A")
+        total = float(quantity) * float(price) if quantity and price else "N/A"
+        products.append(
+            {
+                "Product": product,
+                "Quantity": quantity,
+                "Price": price,
+                "Total": total,
+            }
+        )
+    return products
+
+# Save products to Excel
+def save_to_excel(products, file_path):
+    df = pd.DataFrame(products)
+    df.to_excel(file_path, index=False)
+
+# Celery task for Excel generation
+@celery.task
+def generate_excel_task():
+    products = fetch_products()
+    file_path = "products_data.xlsx"
+    save_to_excel(products, file_path)
+    print(f"Excel file generated at {file_path}")
+
+# FastAPI route: Home page
+@app.get("/")
+async def index(request: Request):
+    files = [f for f in os.listdir(".") if f.endswith(".xlsx")]
+    return templates.TemplateResponse("index.html", {"request": request, "files": files})
+
+# FastAPI route: Download file
+@app.get("/download/{filename}")
+async def download_file(filename: str):
+    file_path = os.path.join(".", filename)
+    return FileResponse(file_path, media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
+
+# FastAPI route: Generate Excel
+@app.get("/generate_excel")
+async def generate_excel():
+    generate_excel_task.delay()  # Trigger Celery task
+    return {"message": "Excel generation started!"}