Update main.py

main.py

@@ -76,6 +76,25 @@ def process_receipt(model, image):
     except Exception as e:
         st.error(f"Gemini error: {str(e)}")
         return "{}"
+
+
+# Write report 
+@app.route('/api/write-report', methods=['POST'])
+def generate_report():
+    prompt = """You are the TrueSpend AI analyst, analyze this transaction data 
+    and write an insightful business report on the spending habits of the users """
+    try:
+        # Get the transaction data from the request body
+        transaction_data = request.get_json()
+        transaction_json_string = json.dumps(transaction_data['transactions'])
+        
+        # Assuming model is defined/imported elsewhere in your Flask app
+        response = model.generate_content([prompt, transaction_json_string])
+        report = response.text
+        return jsonify({"report": report})  # Wrap the report in an object
+        
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
 # ========================================
 # Authentication Endpoints
 # ========================================